Commit Graph

4 Commits

Author SHA1 Message Date
rgrit91
fb9d612081 Memory management refactored 2021-01-08 06:57:10 +00:00
rgrit91
0fc29b40c5 Support for tensors on GPU 2021-01-06 13:20:48 +00:00
rgrit91
32e4b68061 Fix build when CUDA not available 2021-01-05 13:05:16 +00:00
rgrit91
a229aaa6a4 Buffer protocol for torch tensors 2021-01-04 20:53:03 +00:00