Update requirements-cu12.txt

msaroufim · web-flow · commit 3917b40341e7 · 2025-04-29T13:10:36.000-07:00
diff --git a/cuda_core/tests/requirements-cu12.txt b/cuda_core/tests/requirements-cu12.txt
@@ -1,3 +1,4 @@
+--index-url https://download.pytorch.org/whl/cu126
 pytest
 # TODO: remove this hack once cupy has a cp313 build
 cupy-cuda12x; python_version < "3.13"

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+--index-url https://download.pytorch.org/whl/cu126`
`1`	`2`	`pytest`
`2`	`3`	`# TODO: remove this hack once cupy has a cp313 build`
`3`	`4`	`cupy-cuda12x; python_version < "3.13"`