From 85e9ae20e54b8c4de55c47d9e0c4389881d38671 Mon Sep 17 00:00:00 2001 From: cpuhrsch Date: Wed, 23 May 2018 01:54:16 +0000 Subject: [PATCH] Update tbb (#7734) --- aten/src/ATen/native/Distributions.cpp | 8 ++++---- aten/src/ATen/native/native_functions.yaml | 8 ++++---- third_party/tbb | 2 +- tools/autograd/derivatives.yaml | 2 +- 4 files changed, 10 insertions(+), 10 deletions(-) diff --git a/aten/src/ATen/native/Distributions.cpp b/aten/src/ATen/native/Distributions.cpp index 14a8a31dd89e0..1186c76a8be1e 100644 --- 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 index bfeb1bff1faac..0e2d6f79f6fc7 100644 --- 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 index 4c73c3b5d7f78..633b01ad27e01 160000 --- a/third_party/tbb +++ b/third_party/tbb @@ -1 +1 @@ -Subproject commit 4c73c3b5d7f78c40f69e0c04fd4afb9f48add1e6 +Subproject commit 633b01ad27e012e1dc4e392c3230250d1f4967a4 diff --git a/tools/autograd/derivatives.yaml b/tools/autograd/derivatives.yaml index ff8712695d87a..8792233d572f5 100644 --- 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)