ColossalAI/tests/test_fx/test_pipeline/test_hf_model
Frank Lee b2475d8c5c
[fx] fixed unit tests for torch 1.12 (#1327)
2022-07-15 18:22:15 +08:00
..
hf_utils.py [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00
test_albert.py [fx] fixed unit tests for torch 1.12 (#1327) 2022-07-15 18:22:15 +08:00
test_bert.py [fx] fixed unit tests for torch 1.12 (#1327) 2022-07-15 18:22:15 +08:00
test_gpt.py [fx] fixed unit tests for torch 1.12 (#1327) 2022-07-15 18:22:15 +08:00
test_opt.py [fx] fixed unit tests for torch 1.12 (#1327) 2022-07-15 18:22:15 +08:00
test_t5.py [fx] fixed unit tests for torch 1.12 (#1327) 2022-07-15 18:22:15 +08:00