ColossalAI/tests/test_fx/test_tracer
YuliangLiu0306 19438ea0ef
[hotfix] skip gpt tracing test (#2064)
2022-12-02 16:48:28 +08:00
..
test_hf_model [hotfix] skip gpt tracing test (#2064) 2022-12-02 16:48:28 +08:00
test_timm_model [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_torchaudio_model [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_torchrec_model [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_torchvision_model [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_activation_checkpoint_annotation.py
test_bias_addition_module.py
test_control_flow.py
test_functional_conv.py
test_patched_module.py
test_patched_op.py