Commit Graph

38 Commits

Author SHA1 Message Date
Goncalves Henriques, Andre (UG - Computer Science)
22e75becf0 Try to add retain grad 2024-04-21 15:30:33 +01:00
Goncalves Henriques, Andre (UG - Computer Science)
9257404edd Move the name of the module 2024-04-21 15:15:00 +01:00
pjongy
0e95225655 Fix concurrency bug in generating tensor name
TensorCount should be wrapped by lock
- Tensor name came from {PREFIX}_{CURRENT_TENSOR_COUNT}
- atomic Addint64 is good but generating name from this is quite bug
  prone
2023-10-23 13:46:08 +00:00
sugarme
e0fce090d5
Merge pull request #112 from pjongy/minor_cleanup_memory
MINOR: Move runtime.SetFinalizer into freeXXX
2023-10-19 16:10:39 +11:00
pjongy
6f7d68572b Fix build error 2023-10-19 04:18:26 +00:00
pjongy
8031bb85f5 MINOR: Move runtime.SetFinalizer into freeXXX 2023-10-19 04:15:16 +00:00
pjongy
f2b5c911d3 Fix same but in scalar 2023-10-19 04:10:42 +00:00
pjongy
7533f43c2e Fix concurrency issue in newTensor 2023-10-19 03:17:21 +00:00
sugarme
bf24e57df9 upgrade libtorch2.1.0 2023-10-11 14:53:35 +11:00
sugarme
f0b87eb001 fixed ununique tensor name caused double free tensor error 2023-08-31 22:42:07 +10:00
sugarme
1cffab577c fixed Linear.Forward with bias is nil and clean up 2023-08-12 22:09:34 +10:00
sugarme
b3d821d34e temp use of ts.Randn() instead of ts.Uniform() as it causes mem leak 2023-08-12 20:42:39 +10:00
sugarme
e9278816b2 fixed incorrect APIs generation 2023-08-05 13:02:56 +10:00
sugarme
bdf252d831 added debug stack trace to tensor error 2023-07-28 18:06:53 +10:00
sugarme
3cd8d8560f generated newTensor() for GC collection 2023-07-26 23:19:38 +10:00
sugarme
aa23a1e59b WIP: pytorch v2 2023-07-26 21:43:21 +10:00
sugarme
3959bc3f93 fixed read npy failed with single element tensor with zero shape 2023-07-25 13:58:19 +10:00
sugarme
30068c6b41 added ts 2023-07-23 14:54:02 +10:00
sugarme
1fc13ec55e added tensor IsContiguous and IsMkldnn APIs 2023-07-23 14:08:04 +10:00
sugarme
6aa57478f5 added 2023-07-23 13:10:24 +10:00
sugarme
66599380d4 clean-up 2023-07-23 12:36:55 +10:00
sugarme
d1b9267c77 added missing 2023-07-23 12:17:27 +10:00
sugarme
dbbb4b97c1 added get/set cuda device for qt testing 2023-07-11 15:35:36 +10:00
sugarme
01f1e3a7fc wired up ts.Drop() for directly free mem 2023-07-07 22:30:08 +10:00
sugarme
c1ee7689ad fixed check null at tensor ops return slice of tensors and clean-up 2023-07-07 16:08:15 +10:00
sugarme
523061eca6 reworked gotch.dtype with more dtypes 2023-07-07 00:01:23 +10:00
sugarme
c095d9f7f8 removed CleanUp inside ts.NoGrad() 2023-07-06 00:20:11 +10:00
sugarme
f45f0a7ed0 changed []ts.Tensor -> []*ts.Tensor 2023-07-05 23:56:48 +10:00
sugarme
e29288f63e fetched from master 2023-07-04 23:26:20 +10:00
Tim Cassidy
8a08a1d6ab add Progress() method to TextDataIter 2023-02-10 19:30:09 -08:00
sugarme
494b22fbbd cleanup 2022-05-06 18:21:52 +10:00
sugarme
b6b3973b3b improved ts/README.md 2022-05-06 18:16:32 +10:00
sugarme
670d1e9cdf added Conv2D benchmark 2022-05-06 18:10:38 +10:00
sugarme
7e4799eb52 improved ts.Format() info 2022-04-28 17:39:46 +10:00
sugarme
ce421dd4c5 reworked ts.Format() 2022-04-28 17:31:23 +10:00
sugarme
6397c08813 fixed jit test and regenerated APIs 2022-03-16 12:34:38 +11:00
sugarme
c24abefcf3 upgraded to libtorch-1.11 required cuda11.3 2022-03-13 12:56:11 +11:00
sugarme
1c2e6f2599 change package 'tensor' to 'ts' 2022-03-12 18:20:20 +11:00