fix test

pytorch · May 15, 2024 · d55c86c · d55c86c
1 parent 0305dcf
commit d55c86c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/benchmark_experiment.py b/benchmarks/benchmark_experiment.py
@@ -22,7 +22,7 @@ def list_experiment_configs(self):
         "xla": [None, "PJRT", "XRT"],
         "xla_flags": [None],
         "dynamo": [None, "inductor", "openxla_eval", "openxla"],
-        "torch_xla2": ["extract_jax",
+        "torch_xla2": [None, "extract_jax",
                        "torch_export"],  # options only apply to torch_xla2
         "test": ["eval", "train"],
     }