Fixes an error when running on CPU mode.

#2
by toncho11 - opened
No description provided.

If cpu is used in PyTorch it gives the following error:
RuntimeError: "addmm_impl_cpu_" not implemented for 'Half'

tloen changed pull request status to merged

Sign up or log in to comment