wip bump torch rocm version

Signed-off-by: Austin Liu <[email protected]> wip bump torch rocm version Signed-off-by: Austin Liu <[email protected]>
linkedin · Dec 30, 2024 · 45759a3 · 45759a3
1 parent 2687add
commit 45759a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@ def get_default_dependencies():
         ]
     elif platform == "rocm":
         return [
-            "torch>=2.6.0.dev",
+            "torch>=2.6.0.dev20241122+rocm6.2",
             "triton>=3.0.0",
         ]