Skip to content

Instantly share code, notes, and snippets.

@cpuhrsch
cpuhrsch / -
Created February 20, 2019 01:26
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index d6254e952..db0c05fe9 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -18547,11 +18547,11 @@
with_gil: false
deprecated: false
- name: _cudnn_rnn
- matches_jit_signature: false
+ matches_jit_signature: true
@cpuhrsch
cpuhrsch / -
Created February 22, 2019 18:49
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index ca39830a2..67d6c883c 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -53713,7 +53713,7 @@
with_gil: false
deprecated: false
- name: normal_out
- matches_jit_signature: false
+ matches_jit_signature: true
@cpuhrsch
cpuhrsch / -
Created February 25, 2019 20:15
This file has been truncated, but you can view the full file.
diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
index 56f003488..bcdce9c24 100644
--- a/aten/src/ATen/native/native_functions.yaml
+++ b/aten/src/ATen/native/native_functions.yaml
@@ -4093,7 +4093,7 @@
matches_jit_signature: True
python_module: nn
-- func: elu_backward(Tensor grad_output, Scalar alpha, Scalar scale, Scalar input_scale, Tensor output, *, Tensor(a!) grad_input) -> Tensor(a!)
+- func: elu_backward(Tensor grad_output, Scalar alpha, Scalar scale, Scalar input_scale, Tensor out, *, Tensor(a!) grad_input) -> Tensor(a!)
@cpuhrsch
cpuhrsch / -
Created February 25, 2019 21:28
This file has been truncated, but you can view the full file.
diff --git a/build/aten/src/ATen/CPUByteType.cpp b/build/aten/src/ATen/CPUByteType.cpp
index 266e2ba1e..c0f316787 100644
--- a/build/aten/src/ATen/CPUByteType.cpp
+++ b/build/aten/src/ATen/CPUByteType.cpp
@@ -2444,9 +2444,9 @@ std::tuple<Tensor,Tensor> CPUByteType::_weight_norm_cuda_interface(const Tensor
std::tuple<Tensor,Tensor> CPUByteType::_weight_norm_cuda_interface_backward(const Tensor & grad_w, const Tensor & saved_v, const Tensor & saved_g, const Tensor & saved_norms, int64_t dim) const {
AT_ERROR("_weight_norm_cuda_interface_backward not supported on CPUByteType");
}
-Tensor CPUByteType::_standard_gamma_grad(const Tensor & self, const Tensor & output) const {
+Tensor CPUByteType::_standard_gamma_grad(const Tensor & self, const Tensor & out) const {
@cpuhrsch
cpuhrsch / -
Created February 26, 2019 22:47
This file has been truncated, but you can view the full file.
diff --git a/build/aten/src/ATen/CPUByteType.cpp b/build/aten/src/ATen/CPUByteType.cpp
index d50b87d1e..50c08d9f5 100644
--- a/build/aten/src/ATen/CPUByteType.cpp
+++ b/build/aten/src/ATen/CPUByteType.cpp
@@ -2603,9 +2603,9 @@ Tensor CPUByteType::histc(const Tensor & self, int64_t bins, Scalar min, Scalar
const OptionalDeviceGuard device_guard(device_of(self));
return at::native::_histc_cpu(/* actuals */ self, bins, min, max);
}
-Tensor & CPUByteType::adaptive_avg_pool2d_out(Tensor & output, const Tensor & self, IntArrayRef output_size) const {
+Tensor & CPUByteType::adaptive_avg_pool2d_out(Tensor & out, const Tensor & self, IntArrayRef output_size) const {
@cpuhrsch
cpuhrsch / -
Created February 27, 2019 23:11
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index 7ca5b4659..852e2cade 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -18800,7 +18800,7 @@
- name: _cudnn_init_dropout_state
matches_jit_signature: false
schema_string: aten::_cudnn_init_dropout_state(float dropout, bool train, int dropout_seed,
- TensorOptions options) -> Tensor
+ ScalarType dtype, Layout layout, Device device) -> Tensor
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index 7ca5b4659..e3281b884 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -18798,9 +18798,9 @@
with_gil: false
deprecated: false
- name: _cudnn_init_dropout_state
- matches_jit_signature: false
+ matches_jit_signature: true
diff --git a/torch/csrc/jit/generated/register_aten_ops_0.cpp b/torch/csrc/jit/generated/register_aten_ops_0.cpp
index 5a9c7d601..b8b4d86cc 100644
--- a/torch/csrc/jit/generated/register_aten_ops_0.cpp
+++ b/torch/csrc/jit/generated/register_aten_ops_0.cpp
@@ -601,7 +601,7 @@ RegisterOperators reg({
}
),
Operator(
- "aten::_sparse_coo_tensor_with_dims_and_tensors(int sparse_dim, int dense_dim, int[] size, Tensor indices, Tensor values, *, ScalarType dtype=float, Layout layout=strided, Device device=\"cpu\") -> Tensor",
+ "aten::_sparse_coo_tensor_with_dims_and_tensors(int sparse_dim, int dense_dim, int[] size, Tensor indices, Tensor values, *, ScalarType dtype, Layout layout, Device device) -> Tensor",
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index 7ca5b4659..249b9a9b5 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -25896,8 +25896,8 @@
with_gil: false
deprecated: false
- name: embedding_backward
- matches_jit_signature: false
- schema_string: aten::embedding_backward(Tensor grad, IndexTensor indices, int num_weights,
diff --git a/build/aten/src/ATen/Declarations.yaml b/build/aten/src/ATen/Declarations.yaml
index 7ca5b4659..f6b91b7cc 100644
--- a/build/aten/src/ATen/Declarations.yaml
+++ b/build/aten/src/ATen/Declarations.yaml
@@ -25896,8 +25896,8 @@
with_gil: false
deprecated: false
- name: embedding_backward
- matches_jit_signature: false
- schema_string: aten::embedding_backward(Tensor grad, IndexTensor indices, int num_weights,