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

1596 Commits

Author SHA1 Message Date
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
CHEN Xinsheng 9758b18c7d fix dtype mismatch in `nn.cross_entropy_loss` 2024-07-19 16:42:55 +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
DongYang Li 133307627e Update nn.py 2024-07-08 17:13:32 +08:00
DongYang Li 9983779d7a Merge pull request #551 from fansunqi/Conv1d_sp 
check input shape in jt.nn.Conv1d_sp
 2024-07-08 17:08:33 +08:00
DongYang Li bdd6bb6de5 Merge pull request #550 from fansunqi/Conv1d 
check input shape in jt.nn.Conv1d
 2024-07-08 17:07:57 +08:00
DongYang Li 2b57b2d988 Merge pull request #555 from fansunqi/Dropout2d 
check input shape in nn.Dropout2d
 2024-07-08 17:05:49 +08:00
DongYang Li c669b1219a Merge pull request #556 from fansunqi/zeroPad2d 
check input shape in jt.nn.ZeroPad2d
 2024-07-08 17:05:22 +08:00
JittorRepos 596368ae7c Merge pull request #557 from fansunqi/ReplicationPad2d 
check input shape in jt.nn.ReplicationPad2d
 2024-07-08 17:04:21 +08:00
JittorRepos 98d7c2d0fa Merge pull request #562 from fansunqi/unfold 
check parameter's positive in jt.nn.Unfold
 2024-07-08 17:03:02 +08:00
lidongyang c47549e673 add isin 2024-07-05 18:12:43 +08:00
DongYang Li dcd6c6b2be update version 2024-07-02 20:02:02 +08:00
fansunqi f4d4c9d55c check parameter's positive in jt.nn.unfold 2024-07-01 15:41:24 +08:00
Sunqi Fan c45dac35e6 Merge branch 'Jittor:master' into fold 2024-07-01 12:26:35 +08:00
fansunqi 45ccf3d2ac check parameters' positive in jt.nn.fold 2024-07-01 12:23:03 +08:00
DongYang Li 4196cb8154 update version 2024-06-25 16:49:44 +08:00
范孙奇 f358fb7518 check input shape and scale factor's positiveness in jt.nn.Upsample 2024-06-10 19:27:29 +08:00
范孙奇 969d810f55 resume 2024-06-10 19:26:40 +08:00
范孙奇 78b7cf091b check input shape and scale factor's positiveness in jt.nn.Upsample 2024-06-10 19:25:53 +08:00
范孙奇 c4480b7e3b check input shape in jt.nn.ReplicationPad2d 2024-06-10 19:08:53 +08:00
范孙奇 d31b0a244d check input shape in jt.nn.ZeroPad2d 2024-06-10 19:05:49 +08:00
范孙奇 1fba329474 check input shape in nn.Dropout2d 2024-06-10 17:02:05 +08:00
范孙奇 958708ed60 modify error information 2024-06-10 16:48:27 +08:00
范孙奇 e6e5949765 add stride check in jt.nn.ConvTranspose 2024-06-10 16:45:53 +08:00
范孙奇 2266d21a8b remove 3D(unbatch) description 2024-06-10 16:43:22 +08:00
范孙奇 db8fcb33da modify stride positive check in jt.nn.conv_transpose; add input shape check in jt.nn.conv_transpose 2024-06-10 16:39:05 +08:00
范孙奇 baf6b45cf1 add input shape check in jt.nn.transpose3d 2024-06-10 16:29:58 +08:00
范孙奇 8fd834465c modify stride positive check in jt.nn.transpose3d 2024-06-10 16:08:48 +08:00
范孙奇 ae0e52dca5 check input shape in jt.nn.ConvTranspose 2024-06-06 20:55:46 +08:00
范孙奇 d895cb9d36 jt.nn.Conv1d in_channels and out_channels must be positive 2024-06-06 20:39:10 +08:00
范孙奇 b4155d8021 jt.nn.Conv1d_sp in_channels and out_channels must be positive 2024-06-06 20:35:05 +08:00
范孙奇 a42198705b check input shape in jt.nn.Conv1d_sp 2024-06-06 20:25:29 +08:00
范孙奇 4d11325634 check input shape in jt.nn.Conv1d 2024-06-06 20:18:31 +08:00
范孙奇 7f6beb58b9 check input1 and input2 shape in jt.nn.Bilinear() 2024-06-06 20:04:35 +08:00
范孙奇 2f11e3bbbe check target shape and output shape in jt.nn.binary_cross_entropy_with_logits 2024-06-06 18:02:29 +08:00
lidongyang 393684f196 polish nn.Sequential attribute 2024-06-05 22:31:20 +08:00
Hanyuxuan f20ea9dcf1 fix illegal parameters of ConvTranspose and Pool,issue #478,#480,#481,#482,#483 2024-05-31 15:23:42 +08:00
Hanyuxuan 26963bc70f fix Pad2d with illegal padding,issue #464,#465,#466,#467 2024-05-31 14:50:47 +08:00
Hanyuxuan 64c6400070 check x.shape and kernel_size of Pool and Pool3d,issue #461,#463 2024-05-31 14:33:07 +08:00
Hanyuxuan c7e31604c2 fix illegal parameters of PixelShuffle of issue #458,fix validity of concat of issue #459 2024-05-30 19:03:17 +08:00
Hanyuxuan 981f60c381 fix illegal parameters of Conv2d issue #471,#472,#473,#474,#475,#476,#477 2024-05-30 16:10:06 +08:00
Hanyuxuan 102a689fee fix illegal parameters of Pool and Pool3d of issue #451,#453,#456,#457 2024-05-30 15:19:19 +08:00
Hanyuxuan 681174a606 a ValueError fix of issue #450 2024-05-30 14:29:34 +08:00
Hanyuxuan fe8fb30136 a IndexError fix of issue #448 2024-05-30 14:00:27 +08:00
Hanyuxuan 317defa7a1 fix: jt.Var.expand with valid index -1 2024-05-29 11:05:53 +08:00
zhc7 fce14c8d9d fix: a minimal quick fix for issue #544 2024-05-22 11:08:28 +08:00
Yi Zhang 2c1a5e14f1 Merge branch 'master' into master 2024-05-21 12:48:58 +08:00
DongYang Li f8fde94c3f Update version to 1.3.9.8 2024-05-20 21:43:48 +08:00
DongYang Li 4e41e6b070 Merge pull request #539 from fansunqi/issue523_branch 
fix issue 523;update jt.nn.Conv1d/Conv3d/conv2d/conv3d
 2024-05-20 21:42:16 +08:00
DongYang Li 21eaa919b9 Merge pull request #540 from fansunqi/issue522_branch 
fix issue 522,520,519,516; update jt.Pool/Pool3d
 2024-05-20 21:41:52 +08:00
DongYang Li f3744aa47d Merge pull request #541 from fansunqi/issue521_branch 
fix issue 521;update jt.nn.MaxUnpool2d/MaxUnpool3d
 2024-05-20 21:41:34 +08:00
lidongyang 9190180d8d polish rocm support 2024-05-20 21:34:26 +08:00
范孙奇 1bd014ca9b fix issue 521;update jt.nn.MaxUnpool2d/MaxUnpool3d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-17 11:34:46 +00:00
范孙奇 e4981653e3 fix issue 522;update jt.Pool/Pool3d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-17 11:22:12 +00:00
范孙奇 dd9ac69eec fix issue 523;update jt.nn.Conv1d/Conv3d/conv2d/conv3d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-17 09:35:12 +00:00
uyzhang 871ed92fc4 fix: Add conditional import for change_function in __init__.py 2024-05-16 16:03:09 +08:00
DongYang Li 4efbbbf75c Merge pull request #443 from co63oc/patch-1 
Update mnist.py
 2024-05-16 15:20:07 +08:00
DongYang Li 9943ddf8de Merge pull request #536 from fansunqi/issue528_branch 
fix issue 528;update conv_transpose
 2024-05-16 15:19:19 +08:00
DongYang Li 5934b20720 Merge pull request #535 from fansunqi/issue529_branch 
fix issue 529;update contrib.argmax_pool()
 2024-05-16 15:18:14 +08:00
DongYang Li 9370896b35 Merge pull request #537 from fansunqi/issue527_branch 
fix issue 527,526;update jt.zeros/ones/full/randn/randint/random
 2024-05-16 14:47:53 +08:00
DongYang Li ee3c68ce7a Merge pull request #538 from fansunqi/issue525_branch 
fix issue 525;update jt.nn.Reflection2d/Replication2d
 2024-05-16 14:43:28 +08:00
DongYang Li 136a710775 polish PixelShuffle in nn.py 2024-05-16 14:35:21 +08:00
DongYang Li 75429f83b7 Merge pull request #534 from fansunqi/master 
fix issue 531,530;update jt.nn.PixelShuffle/jt.histc
 2024-05-16 14:27:24 +08:00
514flowey 96fda6dee5 Merge pull request #518 from 514flowey/complex 
add complex matmul, inv, qr, eig, and svd
 2024-05-16 13:38:53 +08:00
范孙奇 82595dc766 fix issue 525;update jt.nn.Reflection2d/Replication2d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 13:49:26 +00:00
范孙奇 69fc229912 fix issue 526;update jt.randn/random/randint 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 13:34:06 +00:00
范孙奇 9f4c156e12 fix issue 527;update jt.zeros/ones/full 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 13:23:30 +00:00
范孙奇 9baccaed4d fix issue 528;update conv_transpose 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 12:21:21 +00:00
范孙奇 fc252af9a2 fix issue 529;update contrib.argmax_pool()-2 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 12:04:38 +00:00
范孙奇 72f72900d6 fix issue 529;update contrib.argmax_pool() 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 12:01:08 +00:00
范孙奇 0a1a8d738b fix issue 530; update jt.histc 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 11:31:28 +00:00
范孙奇 ab39f8283c fix issue 531 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 11:14:24 +00:00
uyzhang d587961209 Update ACL library and fix bugs in ACL integration 2024-05-15 19:03:04 +08:00
514flowey 290f2aec60 add complex matmul, inv, qr, eig, and svd 2024-05-08 21:09:07 +08:00
DongYang Li 4808184bee
Update version to 1.3.9.6 2024-05-07 10:33:25 +08:00
lidongyang 4c048d185a polish stack 2024-05-06 14:44:14 +08:00
DongYang Li ce783d2950
Merge branch 'Jittor:master' into master 2024-05-06 01:38:45 +08:00
lidongyang 6945c035c5 fix stack grad bug 2024-05-06 01:38:10 +08:00
uyzhang 82b3f99f51 Fix memory leak in TempAllocator::free() 2024-04-24 18:17:43 +08:00
lidongyang 3092bba175 polish atomicCAS use 2024-04-23 00:43:53 +08:00
lidongyang 185ae5d7eb add new cuda12.2 for g++11 2024-04-11 00:39:57 +08:00
DongYang Li aab6b0f784
Update version 1.3.9.5 2024-04-10 02:07:32 +08:00
lidongyang 8c8c5007d8 fix fp16 include 2024-04-10 02:03:00 +08:00
514flowey e5bfdfb162 add attention mask 2024-04-08 19:26:47 +08:00
MenghaoGuo af07705726
Merge pull request #502 from LDYang694/master 
merge jdiff
 2024-04-08 19:10:58 +08:00
DongYang Li c9c08f5cde
Update __init__.py 2024-04-08 18:56:08 +08:00
zjp_shadow 4f2fe9c4d1 Update compatibility 2024-04-02 16:38:12 +08:00
zhangjiapeng f045e196c1 Update Load 2024-04-01 18:00:34 +08:00
Dun Liang 2744992946 add fallback_func for static graph compiler 2024-01-03 03:57:14 +08:00
Dun Liang 4b9d777570 add doc 2023-12-29 03:55:56 +08:00
lidongyang 4ae9171578 fix merge conflict 2023-12-28 16:10:57 +08:00
Dun Liang 2a995d6cc4 update version to 1.3.9.2 2023-12-27 23:20:56 +08:00
Dun Liang fa40284b60 Merge branch 'master' of https://github.com/Jittor/jittor 2023-12-27 23:18:17 +08:00
Dun Liang 6e1327ba07 fix vcompiler bug 2023-12-27 15:16:06 +08:00
Dun Liang e260f2b6bd fitten code 2023-12-05 01:34:26 +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 2bd05ce04e fix pyobj memleak 2023-11-29 14:10:16 +08:00
Dun Liang 881f769ba7 add tile and ne 2023-10-06 06:59:55 +08:00
Dun Liang e4272cdd4a polish computing graph liveness 2023-09-25 14:48:05 +08:00
Dun Liang ea9026aa6b fix memleak 2023-09-25 10:43:07 +08:00
lzhengning 6e78553016 remove unnecessary prints 2023-09-18 02:41:27 +08:00
lzhengning d21dfe65e6 Add SkipFirstBatchesSampler 2023-09-18 02:06:49 +08:00
Dun Liang 77a5f37f12 fix random sampler 2023-09-16 01:56:48 +08:00
Dun Liang 47d1483ca8 dont deep copy optim state dict 2023-09-15 13:49:13 +08:00
Dun Liang 40368453e8 polish bfloat16 2023-09-15 13:12:42 +08:00
Dun Liang c485fdc07b madd bf16 2023-09-08 22:43:31 +08:00
Dun Liang 2457c5ce4a mem opt for llm 2023-09-03 02:14:09 +08:00
lzhengning 9c69783f05 Merge branch 'master' of github.com:Jittor/jittor 2023-08-25 18:04:58 +08:00
Dun Liang 88582fa854 add reuse_np_array 2023-08-25 17:50:22 +08:00
zhouwy19 ca05be86f0
Update vcompiler.cc free unused output directly 2023-08-15 18:15:44 +08:00
Dun Liang dea40b30a3 bug fix for getitem 2023-08-10 03:19:57 +08:00
Dun Liang a6fca86633 polish nccl wrapper 2023-08-08 01:03:22 +08:00
Dun Liang c48216fb75 Merge branch 'master' of github.com:Jittor/jittor 2023-08-06 18:27:53 +08:00
Dun Liang 3a65eb3c43 add mpi barrier 2023-08-06 18:26:50 +08:00
yang guo ye 21b551411e
Merge pull request #486 from 514flowey/complex_number 
add complex matmul
 2023-08-06 17:05:52 +08:00
514flowey 14c39a7a53 add complex matmul 2023-08-06 16:58:50 +08:00
Dun Liang aa2dc39317 Merge branch 'master' of github.com:Jittor/jittor into HEAD 2023-08-02 00:15:43 +08:00
Dun Liang f03556533c MPI half type 2023-08-02 00:14:39 +08:00
lzhengning 6df06213e3 polish mpi dataset 2023-08-01 21:35:03 +08:00
Gword 7ea3c5530d save&load 2023-07-27 21:03:05 +08:00
Dun Liang f65c600508 remove print 2023-07-26 14:45:32 +08:00
Dun Liang 54112e0e2a Merge branch 'master' of github.com:Jittor/jittor 2023-07-26 00:35:39 +08:00
Dun Liang 39cff908a7 add lock for mpi compile_module 2023-07-26 00:34:34 +08:00
cxjyxxme 89ef365c1a install cutlass 2023-07-25 15:27:27 +08:00
Dun Liang 3f2d0313ed support detial nan checker 2023-07-24 12:41:24 +08:00
Dun Liang 4b35e70b65 polish code 2023-07-17 13:47:58 +08:00
Dun Liang 9a73639a09 polish acl interface 2023-07-13 16:07:48 +08:00
Dun Liang b8854f93d1 add acl support [version 1.3.8.7] 2023-07-13 14:36:57 +08:00
Dun Liang c99546586d update version to 1.3.8.6 2023-07-07 22:09:15 +08:00
Dun Liang 5b6c26ede7 Merge commit '1018e3b2' into HEAD 2023-07-07 22:08:03 +08:00
Dun Liang 1018e3b237 add save pytorch format 2023-07-07 19:18:51 +08:00
Dun Liang 192a11c6b6 add ccec support 2023-06-30 23:31:34 +08:00
Dun Liang f2fdf89fdc add fp16 acc 2023-06-27 22:22:00 +08:00
li-xl 640bfd9c6c fix save checkpoint bugs 2023-06-26 18:36:48 +08:00
Dun Liang a92051ec23 polish version 3.9 2023-06-25 22:58:20 +08:00
Dun Liang df4e6dd5cf add static compiler 2023-06-25 22:55:42 +08:00
li-xl 100d1bdd12 support scalar in getitem && fix bugs for jtorch 2023-06-19 16:06:19 +08:00
Dun Liang b9e34290c0 Merge branch 'master' of github.com:Jittor/jittor 2023-06-10 02:36:30 +08:00
Haoyang Peng ccbe047257
fix load pth with certain format. (#445) 
* fix load pth with certain format.

* fix load 64bit.

* add comments.
 2023-06-10 09:31:35 +08:00
Dun Liang 4d7c7ee748 add __deepcopy__ for dataset method 2023-06-10 02:34:17 +08:00
Dun Liang 39c09a3bc2 add add_module 2023-06-10 01:16:20 +08:00
Dun Liang d595c8b00f Merge branch 'master' of github.com:Jittor/jittor 2023-06-10 01:01:30 +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
li-xl d4691b6f07 update jittor 2023-06-05 16:40:02 +08:00
Dun Liang 48fd6185b9 add sfrl large block setting 2023-06-05 05:30:13 +08:00
li-xl 125ce3e491 fix small bugs 2023-05-27 22:53:22 +08:00
li-xl 5480280429 fix start grad bugs 2023-05-24 07:21:44 +08:00
li-xl b721d851b5 add transformers support 2023-05-24 06:25:25 +08:00
Dun Liang c527badcf1 polish cuda flag set 2023-05-24 04:37:16 +08:00
co63oc 11e29147db Update mnist.py 2023-05-22 13:18:08 +08:00
Dun Liang b0ba0d682e LLM support 2023-05-22 11:28:08 +08:00
Dun Liang f26fdfea22 LLM training tuning 2023-05-16 23:11:05 +08:00
yang guo ye 2b1647f1c9
Merge pull request #436 from 514flowey/complex_number 
add complex number
 2023-05-16 16:42:11 +08:00
Dun Liang 18f054333d fix grad for torch compatible 2023-05-11 19:30:42 +08:00
514flowey 10d9fb1431 add complex number 2023-05-02 18:20:52 +08:00
Dun Liang 66119cbcb5 update version to 1.3.7.15 2023-05-01 16:06:42 +08:00
Dun Liang 289f43e431 Merge branch 'master' of github.com:Jittor/jittor 2023-05-01 16:04:48 +08:00
Dun Liang dd215b0470 polish windows error msg 2023-05-01 16:04:33 +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 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
xingyuxie ab2e2e11d4 add adan optimizer 2023-04-10 13:59:57 +08:00
Dun Liang e395ae7791 update version with weight norm 2023-04-05 02:16:23 +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 d11d2c816e polish include 2023-04-05 02:12:17 +08:00
GanymedeNil 2b79a2c496
Update swap.cc (#429) 
fix getpid not found
 2023-04-05 21:46:21 +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
Dun Liang 57d5b52b18 Merge branch 'master' of github.com:Jittor/jittor 2023-04-04 17:37:07 +08:00
lzhengning 10be18fbc5 support python3.11 on mac 2023-04-04 22:59:36 +08:00
Dun Liang 0036671536 add mem opt doc 2023-04-04 17:36:56 +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 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
zhouwy19 725615872a add from_torch 2022-12-02 17:21:18 +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
Dun Liang ef55bd378f fix isnan check for int 2022-11-19 10:55:04 +08:00
cxjyxx_me fd5bd4aba9 polish ternary 2022-11-17 08:18:25 -05:00
Dun Liang ec2eef1fd9 add profiler mark for easy range profiling 2022-11-16 16:05:27 +08:00