Commit Graph

120 Commits

Author SHA1 Message Date
Hanyuxuan 5df1673608 fix: jt.Var.expand with valid index -1 2024-05-29 11:05:53 +08:00
zhc7 7714ce31d1
fix: a minimal quick fix for issue #544 2024-05-22 11:08:28 +08:00
范孙奇 0a1a8d738b fix issue 530; update jt.histc
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
2024-05-15 11:31:28 +00:00
lidongyang 4c048d185a polish stack 2024-05-06 14:44:14 +08:00
lidongyang 6945c035c5 fix stack grad bug 2024-05-06 01:38:10 +08:00
zhangjiapeng f045e196c1 Update Load 2024-04-01 18:00:34 +08:00
lidongyang 9f5224819e Merge branch 'master' of github.com:Jittor/jittor 2023-12-04 15:47:49 +08:00
lidongyang 1f30881c6e polish GroupNorm&fix different op 2023-12-04 15:46:03 +08:00
Dun Liang 881f769ba7 add tile and ne 2023-10-06 06:59:55 +08:00
Dun Liang c485fdc07b madd bf16 2023-09-08 22:43:31 +08:00
Dun Liang b8854f93d1 add acl support [version 1.3.8.7] 2023-07-13 14:36:57 +08:00
Dun Liang 5b6c26ede7 Merge commit '1018e3b2' into HEAD 2023-07-07 22:08:03 +08:00
li-xl 100d1bdd12 support scalar in getitem && fix bugs for jtorch 2023-06-19 16:06:19 +08:00
Dun Liang 2d4a82d5e5 polish mish activation, thanks! 2023-06-10 00:59:39 +08:00
li-xl ae3c480c21 merge 2023-06-05 16:46:50 +08:00
Dun Liang c527badcf1 polish cuda flag set 2023-05-24 04:37:16 +08:00
Dun Liang f26fdfea22 LLM training tuning 2023-05-16 23:11:05 +08:00
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 2e67d0eb8f
add support for weight_norm and histc. (#420)
* add support for weight_norm and histc.

* add docs.

* fix typo.

---------

Co-authored-by: Exusial <Exusial>
2023-04-05 21:59:20 +08:00
Dun Liang a19936f954 new feature: save_mem 2023-04-02 07:17:55 +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
Dun Liang faa5386d82 chatglm optimize v2 2023-03-31 13:07:47 +08:00
Dun Liang 07f8bd4aff Merge branch 'master' of github.com:Jittor/jittor 2023-03-22 19:47:26 +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
Dun Liang 90850d36ad LLM support with jtorch 2023-03-20 22:33:12 +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
Dun Liang 499d3ee99c add triu and tril function 2022-12-04 20:47:38 +08:00
zhouwy19 725615872a add from_torch 2022-12-02 17:21:18 +08:00
Dun Liang ef55bd378f fix isnan check for int 2022-11-19 10:55:04 +08:00
Dun Liang 5bc160b19c dataset and dataloader interface polish 2022-11-16 11:31:55 +08:00
Dun Liang a57de764f6 fix issue #401, improve atan and atan2 2022-10-05 17:21:51 +08:00
Exusial 48997cd2b3 fix linspace zero division. 2022-09-14 16:14:36 +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
root 87f6c4296e Update compilation and Add the test of cuda 2022-08-30 16:28:08 +08:00
root 5ffa2747a8 update the unique 2022-08-23 01:53:51 +08:00
lidongyang c59351f171 polish linspace 2022-08-13 11:49:09 +08:00
lzhengning e66d22f25c fix: wrong cumsum results with the specified dimension is larger than 4096 2022-05-24 17:05:23 +08:00
zhouwy19 228dbd9583 Merge branch 'master' of github.com:Jittor/jittor 2022-04-29 21:21:24 +08:00
zhouwy19 a59d5b7ce6 add knn op 2022-04-29 21:18:22 +08:00
Dun Liang 5e559cf8fe remove cupy require for cumprod 2022-04-23 16:17:14 +08:00
lzhengning cac2b83a1c update copyright to 2022 2022-04-22 18:01:02 +08:00
Dun Liang 0666456a2f v 1.3.3 memory optimization 2022-04-22 15:04:06 +08:00
Dun Liang 9c74699707 optimize concat and split 2022-04-04 22:53:20 +08:00
Dun Liang cfb8f0c4a8 use -O2 for nan check 2022-03-31 20:28:29 +08:00
Dun Liang 3948cba176 add isnan and isinf check 2022-03-31 20:21:35 +08:00
Dun Liang 64e7ee9d63 polish bool reduce 2022-03-30 14:35:49 +08:00