Commit Graph

601 Commits

Author SHA1 Message Date
Goncalves Henriques, Andre (UG - Computer Science)
d21543e88c Try to fix retain 2024-04-21 17:37:41 +01:00
Goncalves Henriques, Andre (UG - Computer Science)
cc72639158 Try to fix retain 2024-04-21 17:24:12 +01:00
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
sugarme
db661550e5
Merge pull request #120 from sugarme/cflags
clean up cgo flags
2023-11-29 20:58:54 +11:00
sugarme
983939a9b8 clean up cgo flags 2023-11-29 20:55:15 +11:00
sugarme
2a3a963aef minor release 2023-10-31 00:12:06 +11:00
sugarme
0454fa93b5
Merge pull request #118 from sugarme/api
fixed incorrect cbool conversion in autogenerated APIs
2023-10-31 00:03:38 +11:00
sugarme
a1c9ac1dc5 fixed incorrect cbool conversion in autogenerated APIs 2023-10-31 00:02:08 +11:00
sugarme
05eb7f538b
Merge pull request #113 from pjongy/fix_concurrency_bug
Fix concurrency bug in generating tensor name
2023-10-24 14:06:49 +11: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
sugarme
96ed6cb348
Merge pull request #111 from pjongy/bugfix_110_concurrency
Fix concurrency issue in newTensor
2023-10-19 16:06:08 +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
cd0565716f Merge branch 'master' of ssh://github.com/sugarme/gotch 2023-10-11 15:46:44 +11:00
sugarme
293bad50ab fixed lib to C++17 2023-10-11 15:46:11 +11:00
sugarme
8a97ca1559
Merge pull request #106 from sugarme/libtorch2.1
upgrade libtorch2.1.0
2023-10-11 14:54:35 +11:00
sugarme
bf24e57df9 upgrade libtorch2.1.0 2023-10-11 14:53:35 +11:00
sugarme
f06e858def clean-up broken badges 2023-10-11 12:34:56 +11:00
sugarme
f2c1321cb0 upgrade Go version 1.20 2023-10-11 12:32:20 +11:00
sugarme
52d0bc412b update installation script 2023-10-11 12:31:45 +11:00
sugarme
4f03dec060 upgrade libtorch v2.0 2023-10-11 12:00:02 +11:00
sugarme
511dd161b0 updated for versioning 2023-10-11 11:56:28 +11:00
sugarme
83394ef093
Merge pull request #105 from sugarme/libtorch2
Libtorch2
2023-10-11 11:30:26 +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
163e625426 fixed mem blow-up due to gradient accumulation when loading model pretrained 2023-08-12 15:46:51 +10:00
sugarme
ef00723027 remove linear.bias init when not required 2023-08-09 16:25:09 +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
ea87e7fa38 mnist fp16 training comparison 2023-07-07 22:50:12 +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
f9cb2f5cc6 Clean up 2023-07-07 13:20:51 +10:00
sugarme
34e87b1302 added dtype option to nn package 2023-07-07 12:24:56 +10:00
sugarme
523061eca6 reworked gotch.dtype with more dtypes 2023-07-07 00:01:23 +10:00
sugarme
640af9d2df upgrade golang.org/x/image@v0.5.0 2023-07-06 00:26:39 +10:00