Update sdp_utils.cpp to have same version check in all places.

anatoliylitv · web-flow · commit 5fd4fbeab516 · 2025-10-31T14:11:13.000-05:00
diff --git a/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp b/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp
@@ -308,7 +308,7 @@ bool check_flash_attention_hardware_support(sdp_params const& params, bool debug
         }
         return false;
     }
-#if AOTRITON_VERSION_MINOR >= 9
+#if AOTRITON_VERSION_CURRENT >= AOTRITON_VERSION_INT(0, 9)
     if (aotriton::isArchExperimentallySupported(stream)) {
       static const bool enable_experimental = c10::utils::check_env("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL") == true;
       if (!enable_experimental) {
@@ -365,7 +365,7 @@ bool check_mem_efficient_hardware_support(sdp_params const& params, bool debug)
         }
         return false;
     }
-#if AOTRITON_VERSION_MINOR >= 9
+#if AOTRITON_VERSION_CURRENT >= AOTRITON_VERSION_INT(0, 9)
     if (aotriton::isArchExperimentallySupported(stream)) {
       static const bool enable_experimental = c10::utils::check_env("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL") == true;
       if (!enable_experimental) {

Original file line number	Diff line number	Diff line change
`@@ -308,7 +308,7 @@ bool check_flash_attention_hardware_support(sdp_params const& params, bool debug`
`308`	`308`	`}`
`309`	`309`	`return false;`
`310`	`310`	`}`
`311`		`-#if AOTRITON_VERSION_MINOR >= 9`
	`311`	`+#if AOTRITON_VERSION_CURRENT >= AOTRITON_VERSION_INT(0, 9)`
`312`	`312`	`if (aotriton::isArchExperimentallySupported(stream)) {`
`313`	`313`	`static const bool enable_experimental = c10::utils::check_env("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL") == true;`
`314`	`314`	`if (!enable_experimental) {`
`@@ -365,7 +365,7 @@ bool check_mem_efficient_hardware_support(sdp_params const& params, bool debug)`
`365`	`365`	`}`
`366`	`366`	`return false;`
`367`	`367`	`}`
`368`		`-#if AOTRITON_VERSION_MINOR >= 9`
	`368`	`+#if AOTRITON_VERSION_CURRENT >= AOTRITON_VERSION_INT(0, 9)`
`369`	`369`	`if (aotriton::isArchExperimentallySupported(stream)) {`
`370`	`370`	`static const bool enable_experimental = c10::utils::check_env("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL") == true;`
`371`	`371`	`if (!enable_experimental) {`