pytorch
diff --git a/‎benchmarks/float8/float8_roofline.py
Lines changed: 1 addition & 1 deletion b/‎benchmarks/float8/float8_roofline.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎torchao/float8/roofline_utils.py renamed to ‎torchao/testing/float8/roofline_utils.py b/‎torchao/float8/roofline_utils.py renamed to ‎torchao/testing/float8/roofline_utils.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,
 )
Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@`
`61`	`61`	`Float8LinearConfig,`
`62`	`62`	`convert_to_float8_training,`
`63`	`63`	`)`
`64`		`-from torchao.float8.roofline_utils import (`
	`64`	`+from torchao.testing.float8.roofline_utils import (`
`65`	`65`	`get_float8_mem_sympy,`
`66`	`66`	`get_gemm_time_sympy,`
`67`	`67`	`)`