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
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment