[benchmarks] Update DENY_LIST. (#6558)

pytorch · Feb 17, 2024 · f4971a7 · f4971a7
1 parent 3d21839
commit f4971a7
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
@@ -108,8 +108,6 @@
             "accelerator": "tpu"
         },  # The eval test only supports CPU
     ],
-    # self.load_benchmark() exits the main process. See issue #6207.
-    "pytorch_CycleGAN_and_pix2pix": [{}],
     "pyhpc_equation_of_state": [{
         "test": "train"
     },],  # Model's DEFAULT_TRAIN_BSIZE is not implemented
@@ -119,12 +117,6 @@
     "pyhpc_turbulent_kinetic_energy": [{
         "test": "train"
     },],  # Model's DEFAULT_TRAIN_BSIZE is not implemented
-    "pytorch_unet": [
-        {
-            # self.load_benchmark() exits the main process. See issue #6207.
-            "xla": "PJRT",
-        },
-    ],
     "resnet50_quantized_qat": [
         {
             "test": "eval",