From bc4059025fc3d0a606820bc4f77e9d8abba1ba95 Mon Sep 17 00:00:00 2001 From: vasiliy Date: Wed, 26 Feb 2025 09:28:40 -0800 Subject: [PATCH] [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: https://github.com/pytorch/ao/pull/1780 --- benchmarks/float8/float8_roofline.py | 2 +- torchao/{ => testing}/float8/roofline_utils.py | 0 2 files changed, 1 insertion(+), 1 deletion(-) rename torchao/{ => testing}/float8/roofline_utils.py (100%) diff --git a/benchmarks/float8/float8_roofline.py b/benchmarks/float8/float8_roofline.py index 6f30e5eff7..1a428eb80c 100644 --- 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 b/torchao/testing/float8/roofline_utils.py similarity index 100% rename from torchao/float8/roofline_utils.py rename to torchao/testing/float8/roofline_utils.py