mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reenable some test cases that were disabled due to bugs that now seem fixed.

Browse Source 
PiperOrigin-RevId: 684464642
This commit is contained in:
Peter Hawkins 2024-10-10 09:05:13 -07:00 committed by jax authors
parent 19dbff5326
commit 66f526894f
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/BUILD
View File

@ -766,13 +766,6 @@ jax_multiplatform_test(
jax_multiplatform_test(
name = "pytorch_interoperability_test",
srcs = ["pytorch_interoperability_test.py"],
# The following cases are disabled because they time out in Google's CI, mostly because the
# CUDA kernels in Torch take a very long time to compile.
disable_configs = [
"gpu_p100", # Pytorch P100 build times out in Google's CI.
"gpu_a100", # Pytorch A100 build times out in Google's CI.
"gpu_h100", # Pytorch H100 build times out in Google's CI.
],
enable_backends = [
"cpu",
"gpu",
@ -1440,10 +1433,8 @@ jax_multiplatform_test(
"tpu": 20,
},
tags = [
"noasan", # Times out, TODO(b/314760446): test failures on Sapphire Rapids.
"noasan", # Times out
"nodebug", # Times out.
"nomsan", # TODO(b/314760446): test failures on Sapphire Rapids.
"notsan", # TODO(b/314760446): test failures on Sapphire Rapids.
],
deps = [
"//jax:internal_test_harnesses",