Skip to content

Commit 19a65d2

Browse files
committed
Expose stateless methods for torch.cuda.HalfTensor
1 parent 819d4b2 commit 19a65d2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/csrc/cuda/Module.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ static bool THCPModule_assignStateless()
6060
PyObject *stateless;
6161
INIT_STATELESS(Double);
6262
INIT_STATELESS_DETAIL(Float, Cuda);
63+
INIT_STATELESS(Half);
6364
INIT_STATELESS(Long);
6465
INIT_STATELESS(Int);
6566
INIT_STATELESS(Short);

0 commit comments

Comments
 (0)