Commit Graph

84 Commits

Author SHA1 Message Date
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
CHEN Xinsheng 8ab8297430 fix warp (class case) 2024-10-17 21:22:26 +08:00
zjp_shadow 16bfec3a84 fix transpose 2024-10-06 21:20:12 +08:00
uyzhang 94d37047fb polish 2024-10-01 19:34:51 +08:00
uyzhang f5ac2a20b1 Merge branch 'main' of https://github.com/YS-Guo-Lab/JittorHW into main 2024-10-01 18:16:47 +08:00
uyzhang d322ec1673 Refactor transpose_acl function and fix bug in matmul_acl 2024-10-01 18:14:08 +08:00
张仪 68e78de50a format 2024-09-29 13:47:22 +08:00
uyzhang 083aa12e30 Refactor aclnn.h and acl_op.h to add support for FlashAttention and FlashAttentionBackward 2024-09-29 12:29:41 +08:00
uyzhang f9990f0fcd Refactor transpose_acl function and fix bug in matmul_acl 2024-09-27 19:47:13 +08:00
uyzhang e8903d5094 Refactor transpose_acl function and fix bug in matmul_acl 2024-09-27 19:44:04 +08:00