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

1595 Commits

Author SHA1 Message Date
Dun Liang 7a83dcf09e fix windows mkl bug 2023-04-03 07:56:44 +08:00
Dun Liang 16b7966a9a set default amp level 2023-04-03 01:27:55 +08:00
Dun Liang df8628a3a5 Merge branch 'master' of github.com:Jittor/jittor 2023-04-02 07:18:24 +08:00
Haoyang Peng 779274653f
Gamma (#427) 
* add digamma.

* add tdir.
 2023-04-02 13:57:09 +08:00
Dun Liang a19936f954 new feature: save_mem 2023-04-02 07:17:55 +08:00
Dun Liang 00b7783c46 Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 21:14:20 +08:00
Dun Liang e969103399 polish to_float interface 2023-03-31 21:13:59 +08:00
Dun Liang 3300a3e4f3 polish fp16 amp_level3 2023-03-31 21:07:43 +08:00
lzhengning 463fdcc46d rollback version 2023-03-31 19:45:07 +08:00
lzhengning 1119f3cc51 update readme && nn.linear 2023-03-31 19:45:07 +08:00
lzhengning 8a9f0bb904 update load_pytorch.py && add linear 2023-03-31 19:45:07 +08:00
Dun Liang b451a1eb85 Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 13:12:08 +08:00
Dun Liang faa5386d82 chatglm optimize v2 2023-03-31 13:07:47 +08:00
Exusial f79f2d37e8 add digamma. 2023-03-31 11:05:06 +08:00
Dun Liang 1e00883ed1 add attr_dict for jittor var 2023-03-24 01:53:30 +08:00
Dun Liang 5638de9eca LLM support and add version 2023-03-22 19:51:11 +08:00
Dun Liang 07f8bd4aff Merge branch 'master' of github.com:Jittor/jittor 2023-03-22 19:47:26 +08:00
Dun Liang db8679748c fp32_guard && ternary broadcast 2023-03-22 19:41:36 +08:00
lzhengning 50c859f60c support build triu/tril of an inf matrix 
The original implementation results in nan because zeros * inf = nan.
Suggest to use jt.ternary op instead of multiply a mask matrix.
 2023-03-22 19:19:36 +08:00
lzhengning 576b0c9e03 feat: jt.outer & jt.nn.silu 2023-03-21 15:56:21 +08:00
Dun Liang 90850d36ad LLM support with jtorch 2023-03-20 22:33:12 +08:00
lzhengning 2f0e243e01 feat: support dim=None in Var.squeeze to squeeze all dimensions of size 1 2023-03-07 11:33:16 +08:00
LiDongyang 3703fe4421 polish optimzers save&load 2023-01-15 13:38:01 +08:00
Exusial daa0862896 fix multiple sequence load. 2023-01-15 13:37:07 +08:00
Brian Pugh 5dba84ee67 Allow for nn.Sequential to be sliced. 2023-01-15 13:36:29 +08:00
Exusial d1eeec5f89 Add nan fp16 support. 2023-01-15 13:35:10 +08:00
Dun Liang b25e62f1bb polish dataloader 2023-01-15 13:30:03 +08:00
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