From 1ef0c9faba7cc5139a091aa7b667d0985b39a6f5 Mon Sep 17 00:00:00 2001 From: Jagadish Krishnamoorthy Date: Tue, 8 Jul 2025 23:55:38 -0700 Subject: [PATCH] [rocm7.0_internal_testing] Fix test_aot_inductor Signed-off-by: Jagadish Krishnamoorthy --- test/inductor/test_aot_inductor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/inductor/test_aot_inductor.py b/test/inductor/test_aot_inductor.py index cefbaa828b45c..e02447fa1f0f4 100644 --- a/test/inductor/test_aot_inductor.py +++ b/test/inductor/test_aot_inductor.py @@ -30,7 +30,8 @@ from torch.testing._internal.common_cuda import ( SM80OrLater, SM90OrLater, - PLATFORM_SUPPORTS_FLASH_ATTENTION + PLATFORM_SUPPORTS_FLASH_ATTENTION, + PLATFORM_SUPPORTS_FP8 ) from torch.testing._internal.common_device_type import ( _has_sufficient_memory,