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

42 Commits

Author SHA1 Message Date
cxjyxx_me b642b8f1d1 Merge branch 'master' into fft 2022-03-26 23:30:35 -04:00
cxjyxx_me fa62b3a217 support float64 fft 2022-03-26 23:20:16 -04:00
lzhengning de2fceab22 Transitioning from Intel MKL-DNN to oneDNN 2022-03-22 17:10:45 +08:00
Dun Liang 99d6d6b6e1 Merge branch 'master' of github.com:Jittor/jittor 2022-03-21 17:49:24 +08:00
Dun Liang e7bb2545d3 add acl backend 2022-03-21 17:48:50 +08:00
lixl19 1fa89771c4 polish matmul 2022-03-18 16:21:34 +08:00
lixl19 ad57ec890f polish fp16 2022-03-18 12:56:24 +08:00
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
cxjyxx_me e9f681de53 fft 2022-01-25 19:32:06 +08:00
Dun Liang 5b4576c4dd polish tensorcore of cublas in cuda 10 2022-01-10 14:35:30 +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 ac66897047 polish conv transpose group 2021-12-17 17:54:41 +08:00
Dun Liang 9af14f4f55 polish rnn grad 2021-11-22 13:29:04 +08:00
Dun Liang 5cebf93e31 polish rnn grad 2021-11-22 13:16:48 +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 0a2ca63cbc rename wrapper 2021-11-01 12:01:01 +08:00
Dun Liang 08cc8a0451 fix typo 2021-11-01 11:42:38 +08:00
Dun Liang a78c3b4f12 Merge branch 'master' of github.com:Jittor/jittor into win_cuda 2021-10-01 19:16:25 +08:00
Dun Liang 0136d24d83 Merge branch 'win_cuda' of github.com:Jittor/jittor into win_cuda 2021-09-30 14:11:54 +08:00
lzhengning 9ec300f2aa Polish RNN, GRU, and LSTM 
1. Use Cudnn to speed rnn
2. Fix: document string not correctly rendered
3. Fix: RNN cannot accept relu nonlinearity
4. feat: add default hidden state when executing rnn/gru/lstm
 2021-09-28 17:54:56 +08:00
Dun Liang 9293045993 Merge branch 'master' of github.com:Jittor/jittor into win_cuda 2021-09-28 17:23:11 +08:00
Dun Liang 740b9c8552 polish win_cuda in linux 2021-09-28 17:13:03 +08:00
Dun Liang 123e915bb3 support cuda win 2021-09-26 15:35:18 +08:00
Xiang-Li Li 6eb1c77474
Update mpi_broadcast_op.cc 
support uint8 in mpi
 2021-09-25 20:49:29 +08:00
Xiang-Li Li beebddafd5
Update nccl_broadcast_op.cc 
support uint8 in nccl
 2021-09-25 20:48:18 +08:00
Dun Liang 500dfc6ee3 Merge branch 'master' of github.com:Jittor/jittor 2021-09-15 17:45:37 +08:00
Dun Liang c3938e14bf msvc support 2021-09-15 17:34:50 +08:00
Dun Liang 4e712f283f optimize cumsum with unroll 2021-09-02 17:22:46 +08:00
cxjyxx_me 5b830dcf66 cub cumsum 2021-09-02 04:02:31 -04:00
Dun Liang aa749dd0e0 polish exit cleanup 2021-08-31 11:42:12 +08:00
li-xl 317e07907f polish transpose 2021-08-04 16:49:45 +08:00
li-xl aab4bda835 polish dataset 2021-08-04 16:49:45 +08:00
Dun Liang 0748fc1854 polish cutt transpose 2021-07-27 11:13:39 +08:00
Dun Liang ee020b60f7 data gz and register_hook and requires_grads_ 2021-07-26 20:58:15 +08:00
Dun Liang b6fe53e984 optimize conv3d 2021-06-10 15:11:03 +08:00
Dun Liang b721afeac3 support package installed cuda 2021-05-27 19:36:17 +08:00
Dun Liang 6d69be7943 move files 2021-05-16 16:08:31 +08:00
Dun Liang 01b9608f27 remove soft link 2021-05-16 16:04:42 +08:00
Dun Liang 1258121b1f version 1522f3d004f9bdbf3953d91d4c259c341817c71f 2020-03-19 12:20:54 +08:00