diff --git a/hopper/setup.py b/hopper/setup.py
index 95729edabe2..320368af77d 100644
--- a/hopper/setup.py
+++ b/hopper/setup.py
@@ -578,7 +578,7 @@ def nvcc_threads_args():
       
     if torch_version >= target_version:
         flash_api_source = "flash_api_stable.cpp"
-        stable_args = ["-DTORCH_STABLE_ONLY"]  # Checks against including unstable Tensor APIs
+        stable_args = ["-DTORCH_TARGET_VERSION=0x0209000000000000"]  # Targets minimum runtime version torch 2.9.0
     else:
         flash_api_source = "flash_api.cpp"