Skip to content

Commit 24dfd2f

Browse files
authored
[rocm7.0_internal_testing] Fix test_aot_inductor (#2329)
Fixes SWDEV-541804 Added PLATFORM_SUPPORTS_FP8 import alongside existing platform flags. Adjusted import formatting to include a trailing comma. Signed-off-by: Jagadish Krishnamoorthy <[email protected]>
1 parent c3f758e commit 24dfd2f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/inductor/test_aot_inductor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@
3030
from torch.testing._internal.common_cuda import (
3131
SM80OrLater,
3232
SM90OrLater,
33-
PLATFORM_SUPPORTS_FLASH_ATTENTION
33+
PLATFORM_SUPPORTS_FLASH_ATTENTION,
34+
PLATFORM_SUPPORTS_FP8
3435
)
3536
from torch.testing._internal.common_device_type import (
3637
_has_sufficient_memory,

0 commit comments

Comments
 (0)