Commit Graph

2034 Commits

Author SHA1 Message Date
Exusial 8e5ee574f5 merge reduce. 2024-11-27 19:31:48 +08:00
Exusial edf2755cb5 Merge branch 'main' into dev 2024-11-27 11:18:47 +08:00
Exusial 8c33770036 update. 2024-11-27 11:18:19 +08:00
张仪 89f5b98741 split binary and unary op by hy 2024-11-26 23:22:41 +08:00
张仪 e8ae65d797 update 2024-11-26 21:27:06 +08:00
CHEN Xinsheng d4793e2146 Merge pull request #603 from CHEN-Xinsheng/master
fix `nn.Dropout` (dtype convert)
2024-11-26 19:08:17 +08:00
Xinsheng Chen 2bab0bb8dd fix `nn.Dropout` (dtype convert) 2024-11-26 19:07:20 +08:00
CHEN Xinsheng b37fae105b fix `nn.Dropout` (dtype) 2024-11-26 18:59:12 +08:00
Yuxuan Han 8ee6a45d5c Merge pull request #7 from CSCG-Lab/unittest
add isnan, isinf
2024-11-26 10:05:03 +08:00
Yuxuan Han 2580a98710 add isnan, isinf 2024-11-26 10:01:58 +08:00
Yuxuan Han 8a55cfe5b3 Merge pull request #5 from CSCG-Lab/unittest
add stack
2024-11-24 09:47:29 +08:00
Yuxuan Han 66e18b85a8 add stack 2024-11-24 09:46:23 +08:00
张仪 2e137f73b0 big op and recompile bug 2024-11-21 23:59:41 +08:00
Yuxuan Han c8b76acece Merge pull request #4 from CSCG-Lab/unittest
Fix relu grad, flip. Add more unittest
2024-11-21 16:01:35 +08:00
Yuxuan Han d6917eda4c fix relu grad, add more unnittest 2024-11-21 15:59:40 +08:00
Yuxuan Han 0ff1deccb7 fix flip, add softmax 2024-11-21 15:12:12 +08:00
CHEN Xinsheng 8159093262 fix `getitem` (list case) 2024-11-18 20:15:39 +08:00
CHEN Xinsheng d3f2dc5606 add: support numpy int as an index for `getitem` 2024-11-18 17:36:22 +08:00
张仪 47f0c8acda update unit test 2024-11-18 11:54:32 +08:00
CHEN Xinsheng 0b637852f1 Merge pull request #601 from CHEN-Xinsheng/master
fix `stack`
2024-11-14 19:31:45 +08:00
CHEN Xinsheng 1d0602ae32 fix `stack` 2024-11-14 19:30:22 +08:00
CHEN Xinsheng 19b7bbbe57 fix `stack` 2024-11-14 18:01:25 +08:00
CHEN Xinsheng a5b16925e8 add unit test for `any` 2024-11-14 17:35:25 +08:00
CHEN Xinsheng 9b6fd17e20 add `jt.Var.cumsum` and `jt.Var.cub_cumsum` 2024-11-14 17:34:48 +08:00
DongYang Li c10acf34bc Update version 1.3.9.12 2024-11-14 16:17:33 +08:00
CHEN Xinsheng 3d06d25077 add ACL op `any` 2024-11-12 10:32:07 +08:00
CHEN Xinsheng a0dfdc5ff0 add `getitem` (`None` case) 2024-11-11 21:33:55 +08:00
邓一轩 e0537e5c1a concat 2024-11-11 19:58:31 +08:00
邓一轩 58fc5a9b35 fix conv 2024-11-11 19:43:56 +08:00
邓一轩 495a26a458 add sync at end of all op 2024-11-11 17:09:22 +08:00
邓一轩 b9986ac53b use switch 2024-11-11 15:52:35 +08:00
CHEN Xinsheng c747053a54 fix `concat` 2024-11-09 20:13:47 +08:00
514flowey 1474ebe608 Merge pull request #600 from 514flowey/master
update optimizer
2024-11-05 14:18:37 +08:00
514flowey 382bd3f0e5 update optimizer 2024-11-05 14:14:10 +08:00
CHEN Xinsheng 4e4e67dfd5 fix `nonzero` 2024-11-05 12:46:00 +08:00
dengyx21 810af5953b Revert "sync only on broadcast_to from [1]"
This reverts commit 1439a03fca.
2024-11-05 11:29:10 +08:00
CHEN Xinsheng 33bd28fdb3 add ACL op `where` (unary case) 2024-11-04 22:41:16 +08:00
CHEN Xinsheng 19d2e2e912 add ACL op `nonzero`, a temporary implementation, a bit slow 2024-11-04 22:38:29 +08:00
dengyx21 1902dab9c5 sync only on broadcast_to from [1] 2024-11-04 19:42:09 +08:00
CHEN Xinsheng f79e2908ed Merge pull request #3 from CSCG-Lab/jtorch
fix some bugs for jtorch
2024-10-30 12:34:08 +08:00
dengyx21 158ec0756c shut off a stream 2024-10-29 20:14:41 +08:00
CHEN Xinsheng b279960344 Merge branch 'main' 2024-10-29 15:42:39 +08:00
DongYang Li 56bc5f65be fix jt.index error 2024-10-24 02:32:02 +08:00
CHEN Xinsheng f34e1beafa fix warp (class case) 2024-10-21 20:39:15 +08:00
CHEN Xinsheng 1776dd4da9 fix warp (class case) 2024-10-21 17:08:01 +08:00
CHEN Xinsheng 8a31c402de fix `jt.index` 2024-10-18 17:18:06 +08:00
CHEN Xinsheng 811dc241d4 fix `jt.Var.triu_` 2024-10-18 11:13:32 +08:00
CHEN Xinsheng 17048da065 fix finfo & iinfo 2024-10-17 22:00:22 +08:00
CHEN Xinsheng cf4ce2c95e fix finfo bug in jittor 2024-10-17 21:36:00 +08:00
CHEN Xinsheng 89010f5475 fix `Var.triu` & `Var.triu_` 2024-10-17 21:24:19 +08:00