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

103 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 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
Dun Liang 438c8ba692 Merge branch 'master' of github.com:Jittor/jittor 2022-03-15 21:18:08 +08:00
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
Zheng-Ning Liu ca2ec40a59
fix: cumsum not available on CPU 2022-03-02 18:05:35 +08:00
Xiang-Li Li 4383886feb
Update misc.py 
polish cub_cumsum to support ‘range’  item assignment'
 2022-02-22 22:34:14 +08:00
lidongyang 5c1164233e polish make_grid 2022-01-04 16:09:28 +08:00
Dun Liang b913cd8254 polish atan2 2021-12-09 14:41:41 +08:00
Dun Liang 08cc8a0451 fix typo 2021-11-01 11:42:38 +08:00
lzhengning e57f0e4879 updated docs 2021-09-30 16:15:32 +08:00
Dun Liang f807a28e6b add ctcloss 2021-09-08 17:45:17 +08:00
lzhengning 40ce5e85cd polish repeat: support higher dimension 2021-09-07 20:15:33 +08:00
cxjyxx_me 546860e19e fix cumprod 2021-09-02 04:10:17 -04:00
cxjyxx_me 8a2e7a1881 fix 2021-09-02 04:09:29 -04:00
cxjyxx_me 010d306163 fix 2021-09-02 04:05:38 -04:00
cxjyxx_me c53d473d2a fix cumsum 2021-09-02 04:02:22 -04:00
Dun Liang 655362efd7 better memory profiler 2021-09-02 11:20:25 +08:00
cxjyxx_me 7349284663 Merge branch 'master' of github.com:Jittor/jittor 2021-08-26 22:28:06 -04:00
cxjyxx_me b9f97430c6 set different seed for different process&worker 2021-08-26 22:27:50 -04:00