Commit Graph

1568 Commits

Author SHA1 Message Date
lzhengning db81cc938f fix: default to use CUDA when both CUDA and ROCm are installed 2022-10-17 12:09:36 +08:00
Dun Liang 335a2e5c1d support conv 2022-10-13 01:20:14 +08:00
Dun Liang 042c3610a3 improve cuda restart issue 2022-10-10 17:31:53 +08:00
Dun Liang c12549020f Merge branch 'master' of github.com:Jittor/jittor 2022-10-10 17:30:36 +08:00
Dun Liang 524564b763 fix some acl issue 2022-10-07 14:20:18 +08:00
Zheng-Ning Liu 7827c45047
Merge pull request #378 from Exusial/npth
Polish loading weights for PyTorch .pth files
2022-10-06 16:50:35 +08:00
Exusial 17db747ae4 merge. 2022-10-06 16:42:08 +08:00
Dun Liang a57de764f6 fix issue #401, improve atan and atan2 2022-10-05 17:21:51 +08:00
Dun Liang 9a5e7ea6f5 code op support more than 10 args 2022-10-05 16:43:00 +08:00
Exusial b21ee40360 fix wrong transpose. 2022-10-03 18:38:14 +08:00
lzhengning b2fb32aa52 feats: defaults to keep the first dim in nn.Flatten 2022-09-29 22:14:29 +08:00
lidongyang 601101ea44 add new inplace function document 2022-09-26 14:35:45 +08:00
lidongyang 59e49b064d fix trunc_normal_ & add some inplace function 2022-09-26 14:35:45 +08:00
Zheng-Ning Liu 7164b1cc0f Polish document 2022-09-22 00:56:15 +08:00
lzhengning a44c016408 Add a flag 2022-09-20 16:23:23 +08:00
lzhengning 1a91f1dd01 update version to 1.3.5.15:
1. pretty-print jt.Var
2. support area mode in interpolate
3. support reshape empty var with uncertain dimension
2022-09-19 21:46:53 +08:00
lzhengning e050dd6dc9 pretty print jt.Var 2022-09-19 21:44:48 +08:00
Exusial b258cf3a84 add documentation and tests. 2022-09-18 23:41:57 +08:00
Exusial 56255578f9 Add interpolate area support. 2022-09-18 23:41:57 +08:00
lzhengning 05ed6c7e34 fix: support reshape empty var with uncertain dimension 2022-09-16 15:36:33 +08:00
Dun Liang e661f19e20 Merge branch 'master' of github.com:Jittor/jittor 2022-09-16 01:01:20 +08:00
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
Xiang-Li Li 95db8b310e
Merge pull request #389 from Jittor/kldiv
feat: add KLDivLoss in jittor.nn
2022-09-13 21:30:47 +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
root 2c8279ebe8 update installation in README 2022-09-07 15:32:40 +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
Exusial b6c9421a9a fix multiple read. 2022-08-27 11:11:22 +08:00
Exusial e58e0d2247 fix index. 2022-08-27 00:03:34 +08:00
Zheng-Ning Liu 1025d94ffa
Update version 2022-08-25 00:25:21 +08:00
Exusial 880441f193 allclose, add log. 2022-08-25 00:23:28 +08:00
Exusial 125adc7a04 add support for non-zip pytorch pth format. 2022-08-25 00:23:28 +08:00
root 5ffa2747a8 update the unique 2022-08-23 01:53:51 +08:00
Exusial 3b814405ce allclose, add log. 2022-08-19 21:57:28 +08:00
Exusial 92b87448a2 add support for non-zip pytorch pth format. 2022-08-18 14:47:34 +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