Commit Graph

1128 Commits

Author SHA1 Message Date
Dun Liang a8880935a3 polish mem info 2022-04-22 13:51:06 +08:00
Dun Liang 0bb901ed9a polish mem info 2022-04-22 12:49:18 +08:00
Dun Liang 7c90e60424 add nvcc_path= check in install_cuda 2022-04-14 12:22:21 +08:00
Dun Liang 9048f3fd41 polish migrate to cpu 2022-04-07 16:42:22 +08:00
Dun Liang db88d73ed1 polish inplace size 2022-04-05 19:00:23 +08:00
Dun Liang 2ff5eba3a1 polish einsum(phy) 2022-04-05 17:19:14 +08:00
Dun Liang 385ab261e7 Merge branch 'einsum' of https://github.com/Exusial/jittor 2022-04-05 16:56:41 +08:00
Dun Liang 2721e9fb55 polish torch issue 2022-04-05 16:55:01 +08:00
Dun Liang 2901e578dc add where(cond,x,y) alias 2022-04-05 12:42:51 +08:00
Dun Liang 9c74699707 optimize concat and split 2022-04-04 22:53:20 +08:00
Dun Liang 54fb38caed polish fp16 compute 2022-04-03 21:50:00 +08:00
Exusial 1aa77b9feb fix numpy. 2022-04-03 04:18:19 -04:00
Exusial fca1688bf1 Merge branch 'master' of github.com:Exusial/jittor 2022-04-02 22:56:16 -04:00
Exusial 1bd1ca768c add einsum and test. 2022-04-02 22:55:14 -04:00
Dun Liang 18795bd02f polish gopt setitem concat 2022-04-02 23:02:58 +08:00
Dun Liang da0dc2cfba add bug 42 test 2022-04-02 14:19:08 +08:00
Dun Liang 14f0e9be73 polish numpy code op 2022-04-02 14:15:22 +08:00
Dun Liang 0c66be5540 polish fp16 compute without if constexpr 2022-04-01 21:50:01 +08:00
Dun Liang 6ad04632ab add clean graph operation 2022-03-31 21:05:03 +08:00
Dun Liang cfb8f0c4a8 use -O2 for nan check 2022-03-31 20:28:29 +08:00
Dun Liang 3948cba176 add isnan and isinf check 2022-03-31 20:21:35 +08:00
Dun Liang baf9a91e0c add acl test for linear 2022-03-31 20:21:35 +08:00
yang guo ye 52127befec
Merge pull request #303 from Jittor/fix_bug
fix bug of   bool xxop bool
2022-03-31 04:02:41 -05:00
Dun Liang 22948ba07a polish transpose and matmul 2022-03-31 12:40:59 +08:00
cxjyxx_me 6858913536 Merge branch 'master' into fix_bug 2022-03-30 03:09:31 -04:00
cxjyxx_me 4e3f980c0c test bool reduce 2022-03-30 03:08:08 -04:00
cxjyxx_me 4ac5c66b67 bool reduce test 2022-03-30 03:05:36 -04:00
cxjyxx_me 9cdd36a3fa bool binary test 2022-03-30 03:05:30 -04:00
cxjyxx_me 16f9eba32e fix bool 2022-03-30 03:05:15 -04:00
Dun Liang b8c3c82c40 Merge branch 'master' of github.com:Jittor/jittor 2022-03-30 14:36:09 +08:00
Dun Liang 64e7ee9d63 polish bool reduce 2022-03-30 14:35:49 +08:00
Dun Liang 6644cf890b add seperate test 2022-03-30 13:50:14 +08:00
yang guo ye cbcfc57775
Merge pull request #302 from Jittor/fft
Fft
2022-03-30 00:30:12 -05:00
cxjyxx_me 45df1a4eee fft polish 2022-03-30 01:29:25 -04:00
cxjyxx_me 592ead713c fix bool xx bool 2022-03-29 09:45:14 -04:00
cxjyxx_me 65832ac10f clear cache 2022-03-28 09:41:46 -04:00
cxjyxx_me 0b5e367cb1 add cufft plan map 2022-03-28 09:36:10 -04:00
cxjyxx_me aee89141a8 fix bug of uint8.any_() 2022-03-28 07:53:25 -04:00
cxjyxx_me 254b8609fe add cufft wrapper 2022-03-28 07:45:33 -04:00
Dun Liang 2fe3ba696e force unverified ssl context 2022-03-28 14:51:50 +08:00
cxjyxx_me b642b8f1d1 Merge branch 'master' into fft 2022-03-26 23:30:35 -04:00
cxjyxx_me b04ad0ccb4 Merge branch 'master' into fft 2022-03-26 23:22:55 -04:00
cxjyxx_me fa62b3a217 support float64 fft 2022-03-26 23:20:16 -04:00
cxjyxx_me e087a56d86 update fft op test 2022-03-26 22:27:01 -04:00
Dun Liang 65e9ff1265 add amp alias 2022-03-26 15:51:47 +08:00
Dun Liang e13215bfd5 add module float64 2022-03-23 20:41:51 +08:00
Dun Liang 4b0c6a032b change rtol 2022-03-23 20:28:48 +08:00
Dun Liang fbc38c33ab prevent trace py var memleak 2022-03-23 17:29:27 +08:00
Dun Liang 0c1151305e update version 2022-03-22 22:49:47 +08:00
Dun Liang 5062b2d6e6 polish fused cpu and gpu op 2022-03-22 22:48:20 +08:00
Xiang-Li Li 1987728950
Merge pull request #300 from Jittor/jittor_home
add jittor_home
2022-03-22 21:12:39 +08:00
liuruiyang98 5625147244 Update trunc_normal_ with var = var.add(XX) and var = var.clamp(XX) 2022-03-22 20:48:41 +08:00
liuruiyang98 3dd187f34c Update var = var.erfinv() and var = var.multiply(XX) 2022-03-22 20:48:41 +08:00
liuruiyang98 4480351396 Update trunc_normal_ 2022-03-22 20:48:41 +08:00
li-xl aa45703143 add jittor_home 2022-03-22 20:28:43 +08:00
lzhengning 10c51c6823 fix a compatible issue with onednn 2.5 2022-03-22 17:18:31 +08:00
lzhengning de2fceab22 Transitioning from Intel MKL-DNN to oneDNN 2022-03-22 17:10:45 +08:00
Dun Liang 7d9cff9e24 add cgan demo 2022-03-22 16:14:43 +08:00
Dun Liang 6f4e95e55f add simple_cgan demo 2022-03-22 16:14:43 +08:00
SilenceLi 3c8ed89d5e Update __init__.pyi 2022-03-22 15:56:10 +08:00
Dun Liang d4e8092c93 add test_modules 2022-03-22 14:38:50 +08:00
Dun Liang 31fbf4af9c add _modules and _parameters property 2022-03-22 14:37:44 +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 5efb222dd3 add half alias 2022-03-15 22:02:38 +08:00
Dun Liang 438c8ba692 Merge branch 'master' of github.com:Jittor/jittor 2022-03-15 21:18:08 +08:00
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
Dun Liang 15f162d4f3 cuda11.4 in windows disabled 2022-03-10 17:11:55 +08:00
Dun Liang 56af8b392f update version 2022-03-10 14:02:11 +08:00
Dun 70c462502a polish windows encoding support 2022-03-10 05:56:56 +08:00
Dun Liang 495d78ad20 fix windows encoding bugs 2022-03-09 18:23:42 +08:00
Dun Liang 53b377ee7d fix windows encoding error, thanks liduan for reporting this bug 2022-03-08 16:40:24 +08:00
Dun Liang 61ea95b76f polish windows notebook 2022-03-08 15:14:37 +08:00
Dun Liang f796ad0d7f fix some jupyter related bugs 2022-03-07 17:15:17 +08:00
Zheng-Ning Liu ca2ec40a59
fix: cumsum not available on CPU 2022-03-02 18:05:35 +08:00
Dun Liang 7cf6165a10 profiler polish 2022-02-26 16:14:52 +08:00
lzhengning 9c04a01143 fix: improve compatibility of Windows and python 10 2022-02-26 14:48:53 +08:00
Dun Liang c5acaa2474 polish example 2022-02-24 21:21:04 +08:00
Dun Liang 120057640c customize code op compilation flags 2022-02-24 21:08:29 +08:00
Xiang-Li Li 83e4f53af5
Update contrib.py
update concat docs
2022-02-24 18:18:16 +08:00
Xiang-Li Li 43d3a47964
Update __init__.pyi
polish docs of arg_reduce
2022-02-24 17:11:57 +08:00
Zheng-Ning Liu 0c8b2e3bb9
Update compiler.py
fix-bug: explicitly set log_v when querying cuda to avoid wrong output of jittor_utils.
2022-02-23 21:32:32 +08:00
Xiang-Li Li 4383886feb
Update misc.py
polish cub_cumsum to support ‘range’  item assignment'
2022-02-22 22:34:14 +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
lzhengning 8c0f66c638 polish nvcc_flags 2022-02-18 14:55:00 +08:00
yang guo ye e9aca0444c
Update nn.py
fix typo
2022-01-30 11:59:38 +08:00
cxjyxx_me e9f681de53 fft 2022-01-25 19:32:06 +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
Dun Liang 47c2f65749 polish data migrate 2021-10-28 19:59:54 +08:00
Dun Liang b488cc7c61 add erfinv 2021-10-28 17:23:45 +08:00
Dun Liang 26db9872c4 polish error msg 2021-10-28 14:56:41 +08:00
lzhengning 5257284740 update version 2021-10-28 13:44:44 +08:00
lzhengning 205bd011e7 polish embedding 2021-10-28 13:44:08 +08:00
lzhengning a778458831 update version 2021-10-26 20:20:28 +08:00
lzhengning 3147634775 polish 2021-10-26 20:14:17 +08:00
lzhengning ea497ce80a polish gen_pyi 2021-10-26 20:14:17 +08:00
Dun Liang db3b20d886 polish gen pyi 2021-10-26 20:14:17 +08:00
lzhengning b5331c9026 gen pyi files 2021-10-26 20:14:17 +08:00
Dun Liang 6c980c2146 add performance test tutorial 2021-10-25 16:12:27 +08:00
Dun Liang 856f201d9f polish init doc 2021-10-22 16:11:05 +08:00
Dun Liang 60f45b6adf polish init document 2021-10-22 16:05:07 +08:00
Dun Liang 10e28e2151 update res2net50 url 2021-10-19 20:01:20 +08:00
Dun Liang 537487695a polish download error msg 2021-10-19 17:48:14 +08:00
Dun Liang 849fb796ce polish no device found error 2021-10-19 13:10:04 +08:00
Dun Liang e24dd4f14a transpose fit pt 2021-10-18 16:02:02 +08:00
Dun Liang 34a9dab5fb polish error msg 2021-10-18 15:37:02 +08:00
Dun Liang 6c64a20493 update version 2021-10-17 15:58:36 +08:00
Dun Liang 57b4973ac1 polish error msg 2021-10-17 15:58:36 +08:00
Dun Liang b077d6c185 polish cuda install 2021-10-17 15:56:44 +08:00
Dun Liang f65b5df1a6 polish msvc install 2021-10-17 14:26:11 +08:00
Dun Liang 878cb365ac add dependency for mpi ops 2021-10-15 16:22:56 +08:00
Dun Liang f9f02df3d6 polish reduce doc and pass_manager 2021-10-15 12:20:50 +08:00
Zheng-Ning Liu e6e7423631
Merge pull request #272 from Jittor/meminfo
polish meminfo
2021-10-14 16:51:31 +08:00
li-xl 2b16f80289 polish meminfo 2021-10-14 16:43:28 +08:00
Dun Liang eee405669f fix issue #271 2021-10-14 16:36:04 +08:00
Dun Liang 7cfd216372 update version 2021-10-13 20:16:32 +08:00
Dun Liang c05193408a cutt for seperate cache 2021-10-13 20:15:17 +08:00
Dun Liang 102ffa31a5 better cache key 2021-10-13 19:17:19 +08:00
Dun Liang 030b30727d add pip_publish 2021-10-12 14:29:13 +08:00
Dun Liang e0d046f449 polish install from source 2021-10-12 14:29:13 +08:00
lzhengning ca2b4f20d0 polish doc for conv, rnn, and activation 2021-10-11 11:54:02 +08:00
Dun Liang 9e93a671c7 add clean cache script 2021-10-11 11:13:43 +08:00
Dun Liang aba912eb22 polish error message 2021-10-09 16:46:53 +08:00
Dun Liang 2951214deb polish getitem win support 2021-10-09 16:07:48 +08:00
Dun Liang b02b3192bc polish space with path 2021-10-09 09:36:06 +08:00
Dun Liang 66d9df2f82 polish win path order 2021-10-05 19:59:30 +08:00
Dun Liang ad015f13f5 polish win path handle 2021-10-05 19:18:35 +08:00
Dun Liang 2bf1b89c5e polish mkl and cutt install 2021-10-03 22:42:18 +08:00
Dun Liang 7bc620f274 add cuda arch recommend 2021-10-01 20:31:57 +08:00
Dun Liang 6300b0908f lower arch support for 30 2021-10-01 19:56:47 +08:00
Dun Liang 0fa0584fd3 polish files and add cuda arch hint 2021-10-01 19:41:04 +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 2866c64876 update version 2021-10-01 19:15:33 +08:00
Dun Liang 9914e0918b polish msvc crt shared link with -MD 2021-10-01 19:15:27 +08:00
lzhengning 1be676caad Merge branch 'master' of https://github.com/Jittor/jittor 2021-09-30 17:47:57 +08:00
Dun Liang 1ccf45a03d update version 2021-09-30 17:42:58 +08:00
lzhengning e57f0e4879 updated docs 2021-09-30 16:15:32 +08:00
Dun Liang 752a5c62ed polish readme and add test 2021-09-30 15:33:21 +08:00
Dun Liang 2e6c6079cb Merge branch 'win_cuda' of github.com:Jittor/jittor into win_cuda 2021-09-30 15:20:40 +08:00
Dun Liang f43edfb665 polish readme 2021-09-30 15:20:28 +08:00
Dun Liang 28985cfea6 polish event queue 2021-09-30 14:41:40 +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 6938c25e19 polish mac support for win_cuda 2021-09-30 13:42:51 +08:00
Dun Liang cfbadbbfb0 polish windows 2021-09-29 20:03:33 +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 5c2fb6bfce update version 2021-09-28 17:17:20 +08:00
Dun Liang 740b9c8552 polish win_cuda in linux 2021-09-28 17:13:03 +08:00
cxjyxx_me 9ab20a5536 Merge branch 'master' of github.com:Jittor/jittor 2021-09-27 23:21:03 -04:00
cxjyxx_me 36faa8b617 polish shape=1 array fuse 2021-09-27 23:20:20 -04:00
cxjyxx_me 1f51e985d0 polish bool add/substract 2021-09-27 23:19:44 -04:00
Dun Liang ec80112709 polish link problem 2021-09-27 12:12:08 +08:00