ColossalAI/tests/test_fx/test_tracer at dba7e0cfb47d98cf3527b68684251d0dca678fcd - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00

Files

History

Frank Lee 2c8c05675d [fx] patched conv and normalization (#1188 )

2022-06-29 18:58:38 +08:00

..

test_control_flow.py

[fx] supported data-dependent control flow in model tracing (#1185 )

2022-06-29 15:05:25 +08:00

test_patched_module.py

[fx] patched conv and normalization (#1188 )

2022-06-29 18:58:38 +08:00