Convert mul to use opmath_gpu_kernel_with_scalars (#64019)
authorEdward Yang <ezyang@fb.com>
Wed, 1 Sep 2021 00:55:23 +0000 (17:55 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 1 Sep 2021 01:33:30 +0000 (18:33 -0700)
commitb23e4f6086f3174ae66243d77c032aef876c0246
treeb9143d4d237050e930f66f24598a7410ea2412aa
parent0733582087b08dc9b3136768a1e4a704fda5b5bb
Convert mul to use opmath_gpu_kernel_with_scalars (#64019)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64019

Note that previously the functor operated on scalar_t and
this modifies it to operate on opmath_t, but this is not
a problem as half precision was implemented by performing the
compute in float anyway.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS

Reviewed By: ngimel

Differential Revision: D30575282

Pulled By: ezyang

fbshipit-source-id: cc6900ef996e755740afe48f9cb4d0366858dd47
aten/src/ATen/native/cuda/BinaryMulDivKernel.cu