Commit Graph

2034 Commits

Author SHA1 Message Date
514flowey 593519203b Merge pull request #586 from fansunqi/dim
fix dim=3 error
2024-09-05 20:18:02 +08:00
范孙奇 2c141fa996 fix dim=3 error 2024-09-05 20:14:36 +08:00
DongYang Li 4b907d493c Merge pull request #584 from liylo/module
Make forward hook modifiy the inputs and outputs
2024-09-04 16:33:04 +08:00
DongYang Li 79527c40e9 Merge pull request #583 from liylo/func
Add support for block diag function
2024-09-04 16:32:45 +08:00
DongYang Li a1fcd0f337 Merge pull request #503 from 514flowey/attention_mask
add attention mask
2024-09-04 16:19:46 +08:00
DongYang Li 96b97ccf55 Merge pull request #549 from fansunqi/bilinear
check input1 and input2 shape in jt.nn.Bilinear()
2024-09-04 16:16:31 +08:00
DongYang Li 818edc962e Merge pull request #558 from fansunqi/Upsample
check input shape and scale factor's positiveness in jt.nn.Upsample
2024-09-04 16:16:18 +08:00
DongYang Li 60d4f5a2ef Merge pull request #582 from liylo/master
fix load_parameter for Parameterlist issue Jittor#581
2024-09-04 16:15:49 +08:00
lidongyang 30b8a637de remove compatibility 2024-09-04 16:11:51 +08:00
liylo df442516ab forward hooks now could modifiy inputs and outputs 2024-08-28 21:35:12 +08:00
liylo 949c6ed676 init 2024-08-28 21:27:02 +08:00
liylo 1c5519acf2 simple implementation for block diag with proper grad 2024-08-28 21:18:56 +08:00
liylo c8ca6d30eb simple implementation for block diag 2024-08-28 21:13:00 +08:00
liylo ddaf3520e3 fix load 2024-08-28 20:50:37 +08:00
514flowey dc6e888d19 Merge pull request #579 from 514flowey/complex
Add Complex Operators
2024-08-22 12:55:05 +08:00
514flowey 1fbd56bb6d fix unique bug 2024-08-22 12:53:03 +08:00
张仪 b4244090ae first commit 2024-08-21 22:15:12 +08:00
514flowey 822955ac00 add several ffunctions 2024-08-20 15:08:19 +08:00
Yi Zhang c124023085 Merge pull request #567 from Hanyx2021/master
complement of test_aclop
2024-08-12 19:51:07 +08:00
Yuxuan Han 1c0cf4c2e4 complement of test_aclop: error of scatter()-multiple and where() 2024-08-12 19:50:29 +08:00
Yuxuan Han b46264b9f8 complement of test_aclop 2024-08-12 19:28:01 +08:00
Yuxuan Han f353b18472 complement of test_aclop 2024-08-01 16:00:00 +08:00
Yuxuan Han 4deb69c4e5 Merge pull request #1 from Jittor/master
Fixed the BUG of ACL op memory
2024-07-26 21:20:43 +08:00
Yuxuan Han 550ca96a75 complement of test_aclop 2024-07-26 21:16:09 +08:00
张仪 c25ac3a4e8 Fixed the BUG of ACL op memory 2024-07-25 15:54:57 +08:00
hanyx 69b6dd3b42 Merge remote-tracking branch 'upstream/master' 2024-07-24 21:18:52 +08:00
Yi Zhang 496b771211 Update acl_compiler.py 2024-07-24 16:20:02 +08:00
Yi Zhang 29f2fbd853 Update compile_extern.py 2024-07-24 15:43:19 +08:00
张仪 53327feff2 feat: enable ACL optimization in split function 2024-07-24 15:25:10 +08:00
Yi Zhang f2a471c2ec Merge pull request #575 from dengyx21/dev-dyx
FEAT! add floor_int
2024-07-24 15:20:15 +08:00
邓一轩 a755d64f9e FEAT! add floor_int 2024-07-24 15:13:57 +08:00
Yi Zhang 279e4113f3 Update compile_extern.py 2024-07-24 15:01:30 +08:00
Yi Zhang 140b17b824 Update acl_compiler.py 2024-07-24 14:53:04 +08:00
Yi Zhang 67d79a66d4 Merge pull request #572 from dengyx21/dev-dyx
FEAT! add aclop unittest
2024-07-19 17:05:00 +08:00
Yi Zhang 8a9c10d615 Format test_aclop.py 2024-07-19 17:04:42 +08:00
邓一轩 2b12e55447 FEAT! add aclop unittest 2024-07-19 17:01:17 +08:00
Yi Zhang f71c00c3d5 Merge pull request #571 from CHEN-Xinsheng/dev-cross_entropy_loss
fix dtype mismatch in `nn.cross_entropy_loss`
2024-07-19 16:46:41 +08:00
CHEN Xinsheng 9758b18c7d fix dtype mismatch in `nn.cross_entropy_loss` 2024-07-19 16:42:55 +08:00
Yi Zhang 54bc8484e9 Merge pull request #570 from dengyx21/dev-dyx
FEAT! where,scatter,cumsum,gather,flip
2024-07-18 20:08:37 +08:00
邓一轩 8f6563cba9 FEAT! where,scatter,cumsum,gather,flip 2024-07-18 20:04:40 +08:00
lidongyang 121fee583d add no gpu device error 2024-07-12 15:07:35 +08:00
Jiapeng Zhang f7bc197200 fix load bugs
fix load bugs of state
2024-07-10 19:58:07 +08:00
hanyx fa8b332f32 ComplexNumber:polar,view_as_complex,view_as_real 2024-07-09 22:27:28 +08:00
Yi Zhang 3f0814b482 Update acl_compiler.py 2024-07-09 21:48:35 +08:00
张仪 2ae2f1d453 update acl 2024-07-09 19:50:35 +08:00
lidongyang 3b2ca1c2c0 Merge branch 'master' of https://github.com/Jittor/jittor 2024-07-09 14:28:57 +08:00
lidongyang a58c8c7988 polish nn.Sequential __getattr__ 2024-07-09 14:28:17 +08:00
DongYang Li 914cd170b4 Merge pull request #548 from fansunqi/binary_cross_entropy_with_logits
check target and output shape in jt.nn.binary_cross_entropy_with_logits
2024-07-08 17:16:26 +08:00
DongYang Li 6736ce68e3 Merge pull request #553 from fansunqi/conv_transpose3d
modify stride positive check in jt.nn.conv_transpose3d/jt.nn.conv_transpose; add input shape check in jt.nn.conv_transpose3d/jt.nn.conv_transpose
2024-07-08 17:15:38 +08:00
DongYang Li dde745407e Merge pull request #554 from fansunqi/ConvTranspose
check stride positiveness and input shape in jt.nn.ConvTranspose
2024-07-08 17:14:05 +08:00