saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

restore at::Half support for caffe2 SumOp (#41952)

Browse source 
Summary:
PR https://github.com/pytorch/pytorch/issues/40379 added long support but removed at::Half support.  Restore at::Half support.

CC ezyang xw285cornell neha26shah

Pull Request resolved: https://github.com/pytorch/pytorch/pull/41952

Reviewed By: colesbury

Differential Revision: D22720656

Pulled By: xw285cornell

fbshipit-source-id: be83ca7fe51fc43d81bc0685a3b658353d42f8ea
This commit is contained in:
Jeff Daily 2020-07-24 10:46:12 -07:00 committed by Facebook GitHub Bot
parent e9e6cc8c83
commit 2e95b29988
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
caffe2/operators/utility_ops.cu
View file

@ -26,7 +26,7 @@ bool WeightedSumOp<CUDAContext>::RunOnDevice() {
template <>
bool SumOp<CUDAContext>::RunOnDevice() {
return DispatchHelper<TensorTypes<float, int32_t, int64_t>>::call(
return DispatchHelper<TensorTypes<float, at::Half, int32_t, int64_t>>::call(
this, Input(0));
}