Skip to content

Commit d8f3b88

Browse files
Enable pytorch_interoperability_test on GPU
PiperOrigin-RevId: 860270747
1 parent 363082f commit d8f3b88

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1265,7 +1265,7 @@ jax_multiplatform_test(
12651265
srcs = ["pytorch_interoperability_test.py"],
12661266
enable_backends = [
12671267
"cpu",
1268-
# "gpu", # TODO(b/477964663): Broken by PyTorch 2.10.0 upgrade.
1268+
"gpu",
12691269
],
12701270
tags = [
12711271
"noasan", # TODO(b/392599624): torch fails to build.

0 commit comments

Comments
 (0)