Merge branch 'main' into fix/rocm-strix-halo-unified-memory

studio/install_python_stack.py

@@ -28,6 +28,7 @@ if str(_BACKEND_DIR) not in sys.path:
 from backend.utils.wheel_utils import (
     flash_attn_package_version,
     flash_attn_wheel_url,
+    has_blackwell_gpu,
     install_wheel,
     probe_torch_wheel_env,
     url_exists,
@@ -837,10 +838,19 @@ def _flash_attn_install_disabled() -> bool:
 
 
 def _ensure_flash_attn() -> None:
-    if NO_TORCH or IS_WINDOWS or IS_MACOS:
-        return
     if _flash_attn_install_disabled():
         return
+    if NO_TORCH:
+        return
+    if has_blackwell_gpu():
+        _step(
+            "warning",
+            "Skipping flash-attn: Blackwell GPU detected (sm_100+); no compatible prebuilt wheel",
+            _cyan,
+        )
+        return
+    if IS_WINDOWS or IS_MACOS:
+        return
     if (
         subprocess.run(
             [sys.executable, "-c", "import flash_attn"],