Commit Graph

1948 Commits (b3d10db5f1bc58f79e6eeb010b76612aeb299730)
 

Author SHA1 Message Date
oahzxl 774d34f1aa refactor flow search
2 years ago
YuliangLiu0306 550f8f8905
[autoparallel] integrate_gpt_related_tests (#2134)
2 years ago
Ziyue Jiang 59e343328d
[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)
2 years ago
github-actions[bot] 937f404253
Automated submodule synchronization (#2136)
2 years ago
Jiarui Fang 65f56f49e8
[example] gpt demo more accuracy tflops (#2178)
2 years ago
Tongping Liu ab54fed292
[hotfix] add kwargs for colo_addmm (#2171)
2 years ago
Arsmart1 a110933d65
[NFC] fix a typo 'stable-diffusion-typo-fine-tune'
2 years ago
Fazzie-Maqianli 9396a18361
Merge pull request #2174 from ziyuhuang123/main
2 years ago
ziyuhuang123 cf5028363c 'diffusion-typo-change'
2 years ago
アマデウス 622f863291
[hotfix] Jit type hint #2161 (#2164)
2 years ago
Jiarui Fang 27327a4c90
[example] add palm pytorch version (#2172)
2 years ago
Zihao 12e7bcd720
register meta func for rnn (#2159)
2 years ago
oahzxl ded1005667 format code
2 years ago
oahzxl d361d533e8 refactor flow tracer
2 years ago
oahzxl d734529a39 move flow tracer
2 years ago
Boyuan Yao cfe2a9bd90
[autoparallel] memory estimation for shape consistency (#2144)
2 years ago
Jiarui Fang b87496a66b
[hotfix] fix auto policy of test_sharded_optim_v2 (#2157)
2 years ago
YuliangLiu0306 16335cb537
[hotfix] fix aten default bug (#2158)
2 years ago
Jiarui Fang a4b4bb01d6
[example] update vit readme (#2155)
2 years ago
Jiarui Fang 2cfe685b9f
[exmaple] add vit missing functions (#2154)
2 years ago
HELSON a7d95b7024
[example] add zero1, zero2 example in GPT examples (#2146)
2 years ago
YuliangLiu0306 1cce6e36ca
[autoparallel] use metainfo in handler (#2149)
2 years ago
Jiarui Fang 9b39170a5c
[version] 0.1.13 (#2152)
2 years ago
Jiarui Fang e0c01d1db1
Revert "[version] version to v0.1.13 (#2139)" (#2153)
2 years ago
Jiarui Fang 2827f41898
[Gemini] GeminiDPP convert to PyTorch Module. (#2151)
2 years ago
Jiarui Fang bdef9dfdbe
[NFC] remove useless graph node code (#2150)
2 years ago
BlueRum b3f73ce1c8
[Gemini] Update coloinit_ctx to support meta_tensor (#2147)
2 years ago
Jiarui Fang 6ad866b684
[version] version to v0.1.13 (#2139)
2 years ago
oahzxl 9d516fa68f fix layernorm
2 years ago
Zihao a128eec9d5
register aten._convolution.default (#2137)
2 years ago
oahzxl e66a18a0bf optimise search
2 years ago
Jiarui Fang ee287620f0
[Gemini] revert ZeROInitCtx related tracer (#2138)
2 years ago
oahzxl e83e3c6154 update memory estimate
2 years ago
アマデウス 077a66dd81
updated attention kernel (#2133)
2 years ago
github-actions[bot] 484fe62252
Automated submodule synchronization (#2131)
2 years ago
YuliangLiu0306 a3c6924deb
[autoparallel] process size nodes in runtime pass (#2130)
2 years ago
YuliangLiu0306 536560ccc0
[autoparallel] implement softmax handler (#2132)
2 years ago
Jiarui Fang c89c66a858
[Gemini] update API of the chunkmemstatscollector. (#2129)
2 years ago
Jiarui Fang 2938edf446
[Gemini] update the non model data record method in runtime memory tracer (#2128)
2 years ago
Jiarui Fang deee317b0f
[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)
2 years ago
Jiarui Fang 8fac837679
[Gemini] update non model data calculation method (#2126)
2 years ago
Fazzie-Maqianli 6c4c6a0409
Merge pull request #2120 from Fazziekey/example/stablediffusion-v2
2 years ago
Fazzie cea4292ae5 support stable diffusion v2
2 years ago
Jiarui Fang 5efda69735
[Gemini] hotfix the unittest bugs (#2125)
2 years ago
Jiarui Fang 05bb28aacf
[Gemini] mapping of preop timestep and param (#2124)
2 years ago
oahzxl de65e6c3e8 support output
2 years ago
oahzxl cda3e8572a support index dupilictae and update loop
2 years ago
oahzxl 1e0fd11bc1 support check_index_duplicate
2 years ago
github-actions[bot] 764bc16f3e
Automated submodule synchronization (#2123)
2 years ago
oahzxl 8754fa2553 change threshold
2 years ago