Update tbb (pytorch#7734)

flyingcarpet-network · May 23, 2018 · 85e9ae2 · 85e9ae2
1 parent f534339
commit 85e9ae2
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/aten/src/ATen/native/Distributions.cpp b/aten/src/ATen/native/Distributions.cpp
@@ -121,10 +121,10 @@ Tensor bernoulli(const Tensor& self, double p, Generator* gen) {
   return native::bernoulli_(result, p, gen);
 }
 
-Tensor bernoulli(const Tensor& self, Generator* gen) {
+Tensor bernoulli(const Tensor& self) {
   Tensor result = self.type().tensor();
   result.resize_(self.sizes());
-  return native::bernoulli(result, self, gen);
+  return native::bernoulli(result, self, nullptr);
 }
 
 Tensor& bernoulli_(Tensor& self, const Tensor& p_, Generator* gen) {
@@ -159,8 +159,8 @@ Tensor& bernoulli_(Tensor& self, double p, Generator* gen) {
   return native::bernoulli_(self, probs, gen);
 }
 
-Tensor& bernoulli_(Tensor& self, Generator* gen) {
-  return native::bernoulli_(self, 0.5, gen);
+Tensor& bernoulli_(Tensor& self) {
+  return native::bernoulli_(self, 0.5, nullptr);
 }
 
 Tensor _standard_gamma_grad_cpu(const Tensor& self, const Tensor& output) {

diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
@@ -154,15 +154,15 @@
 
 - func: bernoulli(Tensor self, Tensor p, Generator* generator=nullptr) -> Tensor
 
-- func: bernoulli(Tensor self, double p=0.5, Generator* generator=nullptr) -> Tensor
+- func: bernoulli(Tensor self, double p, Generator* generator=nullptr) -> Tensor
 
-- func: bernoulli(Tensor self, Generator* generator=nullptr) -> Tensor
+- func: bernoulli(Tensor self) -> Tensor
 
 - func: bernoulli_(Tensor self, Tensor p, Generator* generator=nullptr) -> Tensor
 
-- func: bernoulli_(Tensor self, double p=0.5, Generator* generator=nullptr) -> Tensor
+- func: bernoulli_(Tensor self, double p, Generator* generator=nullptr) -> Tensor
 
-- func: bernoulli_(Tensor self, Generator* generator=nullptr) -> Tensor
+- func: bernoulli_(Tensor self) -> Tensor
 
 - func: bilinear(Tensor input1, Tensor input2, Tensor weight, Tensor? bias) -> Tensor
   variants: function

diff --git a/third_party/tbb b/third_party/tbb
diff --git a/tools/autograd/derivatives.yaml b/tools/autograd/derivatives.yaml
@@ -138,7 +138,7 @@
   batch1: grad.bmm(batch2.transpose(1, 2)) * alpha
   batch2: batch1.transpose(1, 2).bmm(grad) * alpha
 
-- name: bernoulli(Tensor self, Generator generator)
+- name: bernoulli(Tensor self, double p, Generator generator)
   self: zeros_like(grad)
 
 - name: bmm(Tensor self, Tensor mat2)