From 9aea856115eae29bb77af8be098b1acc1f48c24a Mon Sep 17 00:00:00 2001 From: Edward Yang Date: Thu, 29 Nov 2018 16:01:44 -0800 Subject: [PATCH] Expunge use of type() in Distributions.cpp (#14544) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14544 Modern usage is options(). This doesn't have a functional difference, because all call sites were CPU only (where getting the device index right doesn't matter.) Reviewed By: gchanan Differential Revision: D13258252 fbshipit-source-id: c70f8d618ee9caf37ff2469cceaa439348b6114c --- aten/src/ATen/native/Distributions.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aten/src/ATen/native/Distributions.cpp b/aten/src/ATen/native/Distributions.cpp index ca8e880..9fe38e6 100644 --- a/aten/src/ATen/native/Distributions.cpp +++ b/aten/src/ATen/native/Distributions.cpp @@ -130,7 +130,7 @@ Tensor& bernoulli_tensor_cpu_(Tensor& self, const Tensor& p_, Generator* gen) { THGenerator* generator = get_generator(gen); std::lock_guard lock(generator->mutex); using self_t = scalar_t; - if (p_.type().scalarType() == kDouble) { + if (p_.scalar_type() == kDouble) { auto p = std::get<0>(expand_inplace(self, p_.to(kCPU))); CPU_tensor_apply2( self, p, [generator](self_t& ret_val, double& p_val) { @@ -189,7 +189,7 @@ Tensor _standard_gamma_grad_cpu(const Tensor& self, const Tensor& output) { */ Tensor _s_poisson_cpu(const Tensor& lambda, Generator *gen) { - Tensor ret = at::zeros(lambda.sizes(), lambda.type()); + Tensor ret = at::zeros(lambda.sizes(), lambda.options()); AT_DISPATCH_FLOATING_TYPES(ret.type(), "poisson", [&] { THGenerator* generator = get_generator(gen); std::lock_guard lock(generator->mutex); @@ -203,7 +203,7 @@ Tensor _s_poisson_cpu(const Tensor& lambda, Generator *gen) { } Tensor _s_gamma_cpu(const Tensor& alpha, Generator *gen) { - Tensor ret = at::zeros(alpha.sizes(), alpha.type()); + Tensor ret = at::zeros(alpha.sizes(), alpha.options()); AT_DISPATCH_FLOATING_TYPES(ret.type(), "gamma", [&] { THGenerator* generator = get_generator(gen); std::lock_guard lock(generator->mutex); -- 2.7.4