Commit Graph

905 Commits

Author SHA1 Message Date
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
Dun Liang 7cf6165a10 profiler polish 2022-02-26 16:14:52 +08:00
Dun Liang 7c7b856dcf fix op stat 2022-02-20 21:27:31 +08:00
Dun Liang 58e8c650eb rename common_op_type 2022-02-20 21:15:56 +08:00
Dun Liang 2dd70a7002 check crlf 2022-02-20 21:06:09 +08:00
Dun Liang bcb57086c3 fp16 support, opbytype interface 2022-02-20 20:34:49 +08:00
Dun Liang 81b847e6f0 optimize softmax 2022-02-18 16:45:08 +08:00
Dun Liang 5d191e6247 add nvcc_path check 2022-01-18 16:32:23 +08:00
Dun Liang 6a372f5f4f polish reindex reduce fuse op 2022-01-12 16:43:39 +08:00
Dun Liang 70ccdb1f17 polish loop var pass 2022-01-11 21:03:37 +08:00
Dun Liang 6d1b5e42bc polish fuser 2022-01-11 17:25:00 +08:00
Dun Liang 5b4576c4dd polish tensorcore of cublas in cuda 10 2022-01-10 14:35:30 +08:00
Dun Liang f36693c797 add bool setitem alias and vector_to_tuple 2022-01-10 12:36:28 +08:00
Dun Liang 0c35e1a29b update version 2022-01-04 17:30:37 +08:00
Dun Liang 8f6928315e Merge branch 'master' of github.com:Jittor/jittor 2022-01-04 17:30:14 +08:00
Dun Liang 85ba4898cf add function version conv transpose 2022-01-04 17:29:58 +08:00
tupig-7 5fd65c72cb fix one bug in win10 jittor\src\misc\cpu_math.cc(47): error C2065: 'M_PI': undeclared identifier 2022-01-04 16:10:29 +08:00
liuruiyang98 ef86da29f1 Add init.trunc_normal_ 2022-01-04 16:09:57 +08:00
lidongyang 5c1164233e polish make_grid 2022-01-04 16:09:28 +08:00
Dun Liang f20fa98e44 add use_tensorcore flags 2021-12-30 20:59:48 +08:00
Dun Liang 7b9113f828 support tensorcore 2021-12-30 16:28:21 +08:00
li-xl 928f7ae5be Merge branch 'master' of github.com:jittor/jittor 2021-12-30 11:05:23 +08:00
Dun Liang d6b8349203 polish reindex fuse 2021-12-29 17:50:20 +08:00
Dun Liang 3a31e32a1a add disable lock flags 2021-12-28 12:59:43 +08:00
Dun Liang 8c44329def update version 2021-12-27 13:40:54 +08:00
Dun Liang 0d371ebbe4 add __setattr__ and __getattr__ 2021-12-27 13:40:19 +08:00
cxjyxx_me e19c9aef7a adapt ptrace missed 2021-12-27 10:52:55 +08:00
Dun Liang ac66897047 polish conv transpose group 2021-12-17 17:54:41 +08:00
Dun Liang 922c0d8246 remove some tests 2021-12-17 10:30:14 +08:00
Dun Liang 1f2b100be0 add mpi in py example 2021-12-16 17:37:16 +08:00
Dun Liang c82f5146c9 catch more out of memory error 2021-12-16 16:28:39 +08:00
Dun Liang 9013c70ddf update where document 2021-12-16 16:20:00 +08:00
Dun Liang bc083360e7 test array reindex fuse 2021-12-15 15:38:49 +08:00
Zheng-Ning Liu 03dd698e65
fix-doc: concat example 2021-12-15 10:20:58 +08:00
Dun Liang b913cd8254 polish atan2 2021-12-09 14:41:41 +08:00
Dun Liang 6bc17cb99c polish interpolate 2021-12-06 12:21:18 +08:00
lzhengning eae0357224 doc: fix the wrong example of arg_reduce_op 2021-11-25 11:03:55 +08:00
Dun Liang 9af14f4f55 polish rnn grad 2021-11-22 13:29:04 +08:00
Dun Liang 1e1fe66a30 update version 2021-11-22 13:17:31 +08:00
Dun Liang 5cebf93e31 polish rnn grad 2021-11-22 13:16:48 +08:00
lzhengning 38694a1b6e fix: wrong grad of varslice like [::-1] 2021-11-21 16:50:45 +08:00
Zheng-Ning Liu 4bd0c4a2f6
fix: to_tensor applied on a jt.Var
Allow jt.transform.to_tensor applied on a jt.Var (do nothing but return the inputs).
2021-11-09 23:00:57 +08:00
Dun Liang b3ee4a54e4 polish cu key 2021-11-04 20:51:58 +08:00
Dun Liang 0274c22afd polish cu key 2021-11-04 20:51:31 +08:00
Dun Liang 5695a2e61f polish windows and erfinv 2021-11-04 20:43:54 +08:00
li-xl d08afa840d Merge branch 'master' of github.com:jittor/jittor 2021-11-02 17:45:34 +08:00
li-xl 40ed259665 support tensorcore mode 2021-11-02 17:45:08 +08:00
Dun Liang a1a82f6b2c update version 2021-11-01 12:20:45 +08:00
Dun Liang 0a2ca63cbc rename wrapper 2021-11-01 12:01:01 +08:00
Dun Liang 08cc8a0451 fix typo 2021-11-01 11:42:38 +08:00