Commit Graph

1128 Commits

Author SHA1 Message Date
Dun Liang c0ed98cbd6 polish mem leak problem 2022-09-16 00:59:29 +08:00
uyzhang e0bd748ff1 fix droppath 2022-09-15 16:08:02 +08:00
uyzhang 658ab32bda add droppath in jt.nn 2022-09-14 16:35:38 +08:00
Exusial 48997cd2b3 fix linspace zero division. 2022-09-14 16:14:36 +08:00
lzhengning 46e2dcbefd 1. polish index_op doc; 2. polish pyi files 2022-09-14 16:09:14 +08:00
lzhengning 9d176643a0 feat: add KLDivLoss in jittor.nn 2022-09-13 21:22:51 +08:00
lzhengning 3cce690218 feat: add one_hot in jt.nn 2022-09-07 21:48:53 +08:00
lixl19 83622edc97 polish unique 2022-08-30 20:17:30 +08:00
lixl19 c3efd2bd9d fix unique 2022-08-30 19:31:45 +08:00
Xiang-Li Li b54ce7f51b
Merge pull request #374 from zjp-shadow/zjp
Update the unique (add new features)
2022-08-30 19:20:26 +08:00
root 87f6c4296e Update compilation and Add the test of cuda 2022-08-30 16:28:08 +08:00
Exusial 5eaccf538d add epillsis (numpy version) support. 2022-08-29 14:09:16 +08:00
Exusial 6339f62f56 fix einsum space interpret error. 2022-08-27 22:25:00 +08:00
Zheng-Ning Liu 1025d94ffa
Update version 2022-08-25 00:25:21 +08:00
root 5ffa2747a8 update the unique 2022-08-23 01:53:51 +08:00
Zheng-Ning Liu 89bf097445
Update version 2022-08-17 13:54:59 +08:00
Zheng-Ning Liu ae5988f512 Update cutt_transpose_op.cc 2022-08-17 13:51:55 +08:00
lzhengning 72eb7c76bd fix: cutt tranpose error when input is empty 2022-08-17 13:51:55 +08:00
Dun Liang d2b5c281b2 polish self update issue 2022-08-17 13:51:15 +08:00
Dun Liang 9e599ebf9d polish win bug 2022-08-13 23:41:28 +08:00
Dun Liang 64a30cc6fe add reuse array optimization 2022-08-13 23:26:31 +08:00
Dun Liang 51037cbe9f polish keepdim and keepdims 2022-08-13 14:15:33 +08:00
Dun Liang a8f4a97994 update version 2022-08-13 14:00:30 +08:00
Dun Liang 268bdbea80 polish win space support 2022-08-13 13:47:32 +08:00
lidongyang c59351f171 polish linspace 2022-08-13 11:49:09 +08:00
Dun Liang 19ec9d0a4e polish conda lib conflict 2022-08-09 19:25:51 +08:00
cxjyxx_me 8507ee4330 update version 2022-08-07 07:43:43 -04:00
cxjyxx_me 3dcee91262 Merge branch 'master' into gopt_zero 2022-08-07 07:43:13 -04:00
lzhengning 3d4a9240a2 update version 2022-08-06 01:26:42 +08:00
lzhengning 7b8f51b868 fix: no long assume default parallel compilers 2022-08-06 01:08:39 +08:00
lzhengning 75b77e411f feature: official ROCm support 2022-08-06 00:44:33 +08:00
cxjyxx_me ae9273605d polish gopt 2022-08-04 22:04:56 -04:00
lzhengning 2246519eaf support jt.where (cub & rocprim) 2022-08-01 10:37:49 +08:00
Dun Liang 2c91bc1405 polish transform compose 2022-07-31 20:18:53 +08:00
lzhengning 23c4de4901 fix: unable to launch ROCm after upgrade 2022-07-29 15:36:52 +08:00
cxjyxx_me 767a7d9774 polish cufft 2022-07-27 23:33:01 -04:00
Dun Liang a7496d751c polish load pth 2022-07-28 11:13:51 +08:00
Dun Liang 6f178e8b10 update version 2022-07-28 11:13:51 +08:00
Dun Liang 2efa11997d polish arm support 2022-07-28 11:13:51 +08:00
lzhengning 39a866552b fix: occasional segmentation fault when return value is not provided in c++ lambda function 2022-07-27 01:19:59 +08:00
Letian Li 22881c40f4 Make the function full_like and zeros_like can specify the dtype 2022-07-08 03:02:49 +01:00
Zheng-Ning Liu d1c73d3878
fix: dtype in rand_like is not used 2022-07-06 10:00:42 +08:00
Exusial 0691e2e615 update load function. 2022-07-04 23:01:02 +08:00
Exusial 41aa22a7d3 add pth support. 2022-07-04 23:01:02 +08:00
lzhengning b10178a325 Support AMD ROCm and HIP backend 2022-06-29 00:00:09 +08:00
Dun Liang 569af2a138 polish expr type cast 2022-06-23 21:16:19 +08:00
Zheng-Ning Liu 47d938b2f3
Update post_step document 2022-06-22 19:47:27 +08:00
Exusial 348c0050b9
add dlink. (#334)
* add dlink.

* fix if.

* split if.

* delete useless flags.
2022-06-19 13:55:11 +08:00
Dun Liang 46a03098f9 update version 2022-06-04 23:21:44 +08:00
Dun Liang 05464a8227 Merge branch 'master' of github.com:Jittor/jittor 2022-06-04 23:21:17 +08:00
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