Nigel
/
jittor
forked from jittor/jittor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
1,568 Commits 2 Branches 0 Tags 19 MiB
Commit Graph

1177 Commits

Author SHA1 Message Date
Dun Liang 6fbffb7c62 jtorch compatible 2023-01-06 20:48:04 +08:00
Dun Liang 437a720500 add multinomial operator 2023-01-06 19:39:58 +08:00
Dun Liang 5a4ae74a3a add index_select 2023-01-06 17:08:30 +08:00
lzhengning 2b24d436e3 update version 2023-01-05 11:05:35 +08:00
lzhengning eace68f368 Merge branch 'fix-macOS' 2023-01-05 10:12:07 +08:00
lzhengning 948b9796ad support onednn on macOS with intel chips 2023-01-05 10:11:35 +08:00
lzhengning 5a17e3d265 support builtin download of onednn because homebrew no longer provides onednn 2.x 2023-01-05 09:38:03 +08:00
lzhengning 919e863e55 fix depthwise conv error on ROCm 2023-01-04 14:09:53 +08:00
li-xl cc0e4f120a polish cuda thread 2023-01-03 17:47:25 +08:00
lzhengning 24fe48451a update version to fix Windows CUDA bug 2022-12-25 23:49:47 +08:00
Exusial 35763e6c2c fix windows cuda. 2022-12-25 23:45:52 +08:00
lzhengning 6b7d1a27d4 polish corex_compiler.py 2022-12-17 23:33:50 +08:00
lzhengning d50fe5e754 1. fix compilation failure of fp16_compute.h under rocm platform 
2. add jt.compiler.is_cuda to identify if the current device is CUDA
 2022-12-17 21:36:54 +08:00
Dun Liang 2e74df517b fix zeros_like dtype select 2022-12-09 20:24:04 +08:00
Dun Liang 2a06801681 add cascade setitem test 2022-12-08 17:36:19 +08:00
Dun Liang 7c8cfa2016 polish ones interface 2022-12-07 15:34:40 +08:00
Dun Liang dc7cf4abb0 Merge branch 'master' of github.com:Jittor/jittor 2022-12-05 14:44:07 +08:00
Dun Liang c7e604af1a cascade_setitem v[a][b][c] = x -> v[a,b,c] = x 2022-12-05 14:42:15 +08:00
lzhengning a698c88981 improved compatibility with homebrew libomp >= 15.0.6 on macOS 2022-12-04 22:00:40 +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 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