[benchmarks] Update DENY_LIST. (#6558)

pytorch · Apr 22, 2024 · aea9272 · aea9272
1 parent 75b54ee
commit aea9272
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",