Commit Graph

1550 Commits

Author SHA1 Message Date
Dun Liang c7e604af1a cascade_setitem v[a][b][c] = x -> v[a,b,c] = x 2022-12-05 14:42:15 +08:00
Dun Liang 499d3ee99c add triu and tril function 2022-12-04 20:47:38 +08:00
Dun Liang 44fdc718ab fix assign var in th_mode 2022-12-04 18:22:12 +08:00
Dun Liang 87489d719e add klo bench 2022-12-02 22:37:48 +08:00
zhouwy19 a76c664c65
Merge pull request #411 from uyzhang/master
add ccl
2022-12-02 22:05:26 +08:00
zhouwy19 fe6970efc8
Merge pull request #415 from Jittor/from_torch
add from_torch
2022-12-02 22:00:39 +08:00
zhouwy19 725615872a add from_torch 2022-12-02 17:21:18 +08:00
Dun Liang be8faf4dfc add grad memory check 2022-11-30 13:12:10 +08:00
Dun Liang e014f4f25c polish cuda env setup 2022-11-30 11:05:21 +08:00
Dun Liang 607d13079f add data interface for code op without recompile 2022-11-29 13:42:42 +08:00
Dun Liang f7ba3cab31 polish flag scope for profile_mark 2022-11-20 23:39:12 +08:00
Dun Liang ef55bd378f fix isnan check for int 2022-11-19 10:55:04 +08:00
cxjyxx_me fd5bd4aba9 polish ternary 2022-11-17 08:18:25 -05:00
Dun Liang ec2eef1fd9 add profiler mark for easy range profiling 2022-11-16 16:05:27 +08:00
Dun Liang 5bc160b19c dataset and dataloader interface polish 2022-11-16 11:31:55 +08:00
Dun Liang 1f06bbf22e fix dropout2d 2022-11-13 16:36:45 +08:00
Dun Liang 74932f3c32 add dropout 2d 2022-11-13 16:17:48 +08:00
Dun Liang f4f327bd12 update version 2022-11-07 12:02:35 +08:00
lidongyang b5f03f996b fix pr#355&add unittest 2022-11-07 12:01:23 +08:00
liuruiyang98 13f9eaafc0 Support einops for Jittor 2022-11-07 12:01:23 +08:00
uyzhang de6724692f add ccl 2022-11-06 15:20:54 +08:00
Dun Liang 8c9bfb639d update version 2022-11-04 23:56:20 +08:00
Dun Liang 89998ebc60 polish corex compatible 2022-11-04 23:55:55 +08:00
Dun Liang 20357caf42 update corex backend 2022-11-04 14:45:02 +08:00
Dun Liang 7e40d7831c polish import message 2022-10-27 15:45:39 +08:00
Dun Liang bc7467dbff polish backend compiler 2022-10-26 20:25:14 +08:00
Dun Liang 344e13948c add device id interface 2022-10-26 16:29:00 +08:00
Dun Liang 49a0f8ba43 polish backends 2022-10-26 14:39:38 +08:00
Dun Liang 1ccc22caab update version && fit float64 dtype 2022-10-26 14:31:56 +08:00
Dun Liang 9f36d52575 Merge branch 'master' of github.com:Jittor/jittor 2022-10-26 14:31:17 +08:00
Dun Liang 003cdf6b16 polish backend 2022-10-26 14:29:53 +08:00
lzhengning a4eb197938 fix: polish use_[device] 2022-10-18 00:28:36 +08:00
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