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

1128 Commits

Author SHA1 Message Date
Dun Liang 92c86fde75 fix use_cuda_managed_allocator 2022-06-04 23:21:03 +08:00
Gword d5849ac54b update version 2022-06-02 11:16:56 +08:00
Gword fa935e1a6f getlasterror in cutt_transpose 2022-06-02 11:11:50 +08:00
lzhengning 3835a56b5b fix: jt.sync_al() after load_parameters to stop fusion array & broadcast 2022-05-28 11:38:00 +08:00
Dun Liang bc945bae94 polish win bug 2022-05-27 21:03:34 +08:00
Dun Liang 7de460536c update version, add support for jtorch 2022-05-26 15:00:06 +08:00
Dun Liang 8b1d620b22 Merge branch 'master' of github.com:Jittor/jittor 2022-05-26 14:58:33 +08:00
Dun Liang bfa6cd47ae polish test 2022-05-26 14:56:08 +08:00
Dun Liang 5a1fe489cb update data gz 2022-05-26 14:29:54 +08:00
Dun Liang 31fbabfb6f add th_mode 2022-05-25 17:13:43 +08:00
zhouwy19 a011b8a7ed polish unified memory 2022-05-25 12:19:26 +08:00
lzhengning e66d22f25c fix: wrong cumsum results with the specified dimension is larger than 4096 2022-05-24 17:05:23 +08:00
Zheng-Ning Liu 2ae5e019df feat: add dozens of docs 2022-05-24 15:58:55 +08:00
Dun Liang c5ccdaf330 polish jit key 2022-05-23 14:23:02 +08:00
Dun Liang 91fe1fac85 polish win bug 2022-05-20 14:34:54 +08:00
Dun Liang 643ca5bbb4 polish setitem grad nullptr 2022-05-19 11:34:50 +08:00
Dun Liang ba266fa99c polish log_softmax precision 2022-05-18 15:37:09 +08:00
Dun Liang 4c5ac0fda9 polish JITTOR_HOME env 2022-05-18 12:32:57 +08:00
Your Name 772bdfcdf2 fix abs. 2022-05-16 21:02:56 +08:00
Exusial 24f73cfdf2 fix fp16 abs error. Add cublas acc support. 2022-05-16 21:02:56 +08:00
Gword f946599e9e
fix bug: no recompiler #include"" 2022-05-15 21:36:49 +08:00
li-xl 9b7a184808 polish cross_entropy_loss 2022-05-13 20:04:50 +08:00
Dun Liang c113eabcca add th_mode 2022-05-10 18:04:35 +08:00
Dun Liang 88815e8dd3 add global var support for code op 2022-05-09 16:17:46 +08:00
li-xl 59ec0c54f0 polish keep numpy array in multiple worker 2022-05-08 17:00:29 +08:00
Dun Liang f815a8cd95 polish migrate_all_to_cpu 2022-05-06 14:46:50 +08:00
Dun Liang 16bffae407 update version 2022-05-06 13:14:36 +08:00
Dun Liang 41fef77707 Merge branch 'master' of github.com:Jittor/jittor 2022-05-06 12:59:54 +08:00
Dun Liang 88bb84255f add node_order control execute order 2022-05-06 12:57:47 +08:00
zhouwy19 5dcc1392e4 update test_knn 2022-05-03 15:08:44 +08:00
zhouwy19 228dbd9583 Merge branch 'master' of github.com:Jittor/jittor 2022-04-29 21:21:24 +08:00
zhouwy19 a59d5b7ce6 add knn op 2022-04-29 21:18:22 +08:00
Dun Liang eda20ee15e polish windows cuda bug 2022-04-28 22:30:00 +08:00
lzhengning 34740f9e52 fix: encountered nullptr in RNN 2022-04-27 15:55:17 +08:00
Dun Liang 0280f141e2 FIX_TORCH_ERROR by default 2022-04-26 14:27:05 +08:00
lzhengning a1322782ae improve compatibility with miniforge 2022-04-25 12:48:59 +08:00
Dun Liang 3e6fb4cad8 polish win log memptr 2022-04-24 14:25:14 +08:00
Dun Liang 9d899dcf23 polish reindex memory optimize 2022-04-24 13:57:36 +08:00
Dun Liang 0b13930ed3 add slice broadcast 2022-04-23 19:00:15 +08:00
Dun Liang ab30a15eae update version 2022-04-23 18:09:17 +08:00
Dun Liang 8f017cf57c polish fuser 2022-04-23 18:09:05 +08:00
Dun Liang cc6cb28d46 polish dataset use jittor with cuda 2022-04-23 17:32:12 +08:00
Dun Liang 5e559cf8fe remove cupy require for cumprod 2022-04-23 16:17:14 +08:00
Dun Liang 9e58fac49f polish windows install error 2022-04-22 21:40:58 +08:00
Dun Liang 3d4861265d polish fuser 2022-04-22 21:01:17 +08:00
lzhengning cac2b83a1c update copyright to 2022 2022-04-22 18:01:02 +08:00
lzhengning de8a193e9a add support jt.var / Var.var to compute variance. 
Acknowledgement:
    Thanks fangtiancheng https://discuss.jittor.org/t/topic/193/3 for a demo implementation.
 2022-04-22 17:49:51 +08:00
Letian 9b50f6370c Fix _upload 
- Modify jkey permission
 2022-04-22 17:33:04 +08:00
Dun Liang 8e703b83b1 Merge branch 'master' of github.com:Jittor/jittor 2022-04-22 15:04:53 +08:00
Dun Liang 0666456a2f v 1.3.3 memory optimization 2022-04-22 15:04:06 +08:00
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