lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
2,017 Commits 57 Branches 331 Tags 26 MiB
Commit Graph

112 Commits

Author SHA1 Message Date
Yuxuan Han d51503ec57 update from HuaWei ACL 2025-05-12 15:28:18 +08:00
CHEN Xinsheng e1052067c0 support `jt.any` with argument `dim` 2025-02-18 23:29:05 +08:00
zjp_shadow a5c37bd58e Update HCCL to support multi npus 2025-02-10 16:18:34 +08:00
Exusial 7192a9060c delete acl_op.h 2024-12-24 10:04:44 +08:00
Exusial 981391ea6d fix conv,relu, split expand 2024-12-23 17:24:09 +08:00
Yuxuan Han feeee7b3e6
Merge branch 'main' into splits 2024-12-23 15:57:22 +08:00
Exusial 01d300484d split triu,embedding,batchnorm 2024-12-23 15:49:07 +08:00
Exusial 361ad914c1 polish code 2024-12-23 13:43:21 +08:00
Yi Zhang 9b76354a03
Update acl_compiler.py 2024-12-20 22:44:49 +08:00
Exusial 021ffea7c7 split stack,rope,nantonum 2024-12-19 19:40:37 +08:00
Exusial c833f841b1 split silu,sigmoid,softmax 2024-12-19 10:17:06 +08:00
Exusial c94237da44 split relu,dropout,transpose,flashattention 2024-12-19 09:36:32 +08:00
Exusial 6c274e3566 split where,scatter,floor 2024-12-14 10:36:37 +08:00
Exusial 26563ad413 split cumsum,gather,index 2024-12-13 14:31:29 +08:00
Exusial 3381fd6131 split maxpool,flip,concat 2024-12-12 19:49:06 +08:00
邓一轩 4add5a72e9 splite matmul and bmm from acl_op 2024-12-07 11:16:49 +08:00
Exusial 4e60133d48 add setitem & getitem op 2024-12-06 11:21:55 +08:00
Exusial 05a2ec1f84 add conv_op. 2024-12-04 15:41:38 +08:00
Exusial b283fc9102 fixed the bug of not recompile 2024-12-02 22:24:01 +08:00
CHEN Xinsheng 7eae96dc28 add `jt.Var.isnan` and `jt.Var.isinf` 2024-11-28 22:04:42 +08:00
Exusial 0e5aa99b2a update reduce. 2024-11-28 16:24:07 +08:00
Exusial 53754db9c1 update 2024-11-28 15:45:15 +08:00
Exusial de12a34967 fix compile include aclops.h in aclops 2024-11-28 13:24:47 +08:00
Exusial 1c1fc03ca2 Merge branch 'main' into dev 2024-11-27 11:18:47 +08:00
Exusial dff281411e update. 2024-11-27 11:18:19 +08:00
张仪 ed403c1d5a split binary and unary op by hy 2024-11-26 23:22:41 +08:00
张仪 8518077b89 update 2024-11-26 21:27:06 +08:00
Yuxuan Han a7a9f99f1f add isnan, isinf 2024-11-26 10:01:58 +08:00
Yuxuan Han c427df90ef add stack 2024-11-24 09:46:23 +08:00
张仪 793e830fc4 big op and recompile bug 2024-11-21 23:59:41 +08:00
Yuxuan Han 3623342f68 fix relu grad, add more unnittest 2024-11-21 15:59:40 +08:00
Yuxuan Han d6e12c9e93 fix flip, add softmax 2024-11-21 15:12:12 +08:00
CHEN Xinsheng 2c77e8387d fix `getitem` (list case) 2024-11-18 20:15:39 +08:00
CHEN Xinsheng 36c607ec0e add: support numpy int as an index for `getitem` 2024-11-18 17:36:22 +08:00
CHEN Xinsheng 5bd07acd78 add `jt.Var.cumsum` and `jt.Var.cub_cumsum` 2024-11-14 17:34:48 +08:00
CHEN Xinsheng 984e78c183 add ACL op `any` 2024-11-12 10:32:07 +08:00
CHEN Xinsheng ba64bc4a85 add `getitem` (`None` case) 2024-11-11 21:33:55 +08:00
邓一轩 22db5cea5a concat 2024-11-11 19:58:31 +08:00
邓一轩 f62cd1d218 fix conv 2024-11-11 19:43:56 +08:00
CHEN Xinsheng 3efd42e105 fix `concat` 2024-11-09 20:13:47 +08:00
CHEN Xinsheng b849de1052 fix `nonzero` 2024-11-05 12:46:00 +08:00
CHEN Xinsheng 88478c724d add ACL op `where` (unary case) 2024-11-04 22:41:16 +08:00
CHEN Xinsheng ebc6ba3a01 add ACL op `nonzero`, a temporary implementation, a bit slow 2024-11-04 22:38:29 +08:00
CHEN Xinsheng 163793d808 Merge branch 'main' 2024-10-29 15:42:39 +08:00
CHEN Xinsheng f4016dc5a2 fix warp (class case) 2024-10-21 20:39:15 +08:00
CHEN Xinsheng 327a8c7ff9 fix warp (class case) 2024-10-21 17:08:01 +08:00
CHEN Xinsheng 7fb3a6bf8d fix `jt.index` 2024-10-18 17:18:06 +08:00
CHEN Xinsheng 091ea73998 fix `jt.Var.triu_` 2024-10-18 11:13:32 +08:00
CHEN Xinsheng 8c395656bd fix `Var.triu` & `Var.triu_` 2024-10-17 21:24:19 +08:00
CHEN Xinsheng 162107fd7a add cub_cumsum & cumprod 2024-10-17 21:23:32 +08:00