From 5cfb47e377ae7164aa7bc68b2e749c4dcccc0e93 Mon Sep 17 00:00:00 2001 From: zpcore Date: Wed, 15 May 2024 22:04:49 +0000 Subject: [PATCH] fix test --- benchmarks/benchmark_experiment.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/benchmarks/benchmark_experiment.py b/benchmarks/benchmark_experiment.py index 856fac3f0bc..26a56c49f24 100644 --- a/benchmarks/benchmark_experiment.py +++ b/benchmarks/benchmark_experiment.py @@ -22,8 +22,7 @@ def list_experiment_configs(self): "xla": [None, "PJRT", "XRT"], "xla_flags": [None], "dynamo": [None, "inductor", "openxla_eval", "openxla"], - "torch_xla2": [None, "extract_jax", - "torch_export"], # options only apply to torch_xla2 + "torch_xla2": [None], # options only apply to torch_xla2 "test": ["eval", "train"], }