[2/x] float8 cleanup: move roofline utils to testing

Summary: Moves `torchao/float8/roofline_utils.py` to `torchao/testing/float8`, as this is only needed for tests/benchmarks. Test Plan: ``` python benchmarks/float8/float8_roofline.py ~/local/tmp/20250223_test ``` Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 8bfadd4a6a6e96d0d2c3175aa94279ac015fe18e ghstack-comment-id: 2683402019 Pull Request resolved: #1780
pytorch · Feb 26, 2025 · bc40590 · bc40590
1 parent 7d87946
commit bc40590
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/benchmarks/float8/float8_roofline.py b/benchmarks/float8/float8_roofline.py
@@ -61,7 +61,7 @@
     Float8LinearConfig,
     convert_to_float8_training,
 )
-from torchao.float8.roofline_utils import (
+from torchao.testing.float8.roofline_utils import (
     get_float8_mem_sympy,
     get_gemm_time_sympy,
 )

diff --git a/torchao/float8/roofline_utils.py → torchao/testing/float8/roofline_utils.py b/torchao/float8/roofline_utils.py → torchao/testing/float8/roofline_utils.py