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

579 Commits

Author SHA1 Message Date
lzhengning b824ab83c0 update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
Haoyang Peng 4825cead21
add Gamma Distribution on Cuda (#435) 
* add digamma.

* add tdir.

* add gamma distribution.

* add test for gamma distribution.

* update location.

* add api directly into jt.__init__
 2023-04-25 14:25:10 +08:00
lzhengning 6bf14ee650 pip publish for ADAN optimizer, related PR https://github.com/Jittor/jittor/pull/430 2023-04-10 14:03:04 +08:00
Dun Liang e395ae7791 update version with weight norm 2023-04-05 02:16:23 +08:00
Dun Liang d11d2c816e polish include 2023-04-05 02:12:17 +08:00
Dun Liang 0464661920 update version 2023-04-04 23:49:10 +08:00
Dun Liang 45f617cb54 add pid location for swap 2023-04-04 23:15:42 +08:00
Dun Liang 5cf9c10c92 fix windows virtual env bug 2023-04-04 19:30:22 +08:00
Dun Liang 3bcdfc97a9 update version 2023-04-04 17:37:24 +08:00
lzhengning 10be18fbc5 support python3.11 on mac 2023-04-04 22:59:36 +08:00
Dun Liang 33b091e041 support python3.11 2023-04-04 14:54:40 +08:00
Dun Liang b88ffdbdf5 fix windows load pth bug 2023-04-04 13:12:32 +08:00
Dun Liang 2ea015061c polish swap iter 2023-04-03 12:55:01 +08:00
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 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 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
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
lzhengning 2b24d436e3 update version 2023-01-05 11:05:35 +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
lzhengning 6b7d1a27d4 polish corex_compiler.py 2022-12-17 23:33:50 +08:00
Dun Liang 2e74df517b fix zeros_like dtype select 2022-12-09 20:24:04 +08:00
Dun Liang 7c8cfa2016 polish ones interface 2022-12-07 15:34:40 +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
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
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