update

Author: shink

.github/workflows/ascend_npu_test.yml

@@ -95,7 +95,6 @@ jobs:
     name: Build torch_npu
     needs:
       - prepare
-    if: ${{ github.event_name == 'repository_dispatch' }}
     uses: ./.github/workflows/_ascend_npu_build.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
@@ -106,7 +105,6 @@ jobs:
     needs:
       - prepare
       - build
-    if: ${{ github.event_name == 'repository_dispatch' }}
     uses: ./.github/workflows/_ascend_npu_ut.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
@@ -120,7 +118,6 @@ jobs:
       - prepare
       - build
       - test
-    if: ${{ github.event_name == 'repository_dispatch' }}
     uses: ./.github/workflows/_ascend_npu_benchmark.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}
@@ -130,6 +127,7 @@ jobs:
     secrets:
       pr-token: ${{ secrets.COSDT_BOT_TOKEN }}
 
+  # TODO
   pr-comment:
     name: Comment report on upstream PRs
     needs:

requirements.txt

@@ -1 +1,2 @@
+git+https://github.com/pytorch/benchmark.git
 tabulate

src/benchmark/utils.py

@@ -1,21 +1,9 @@
 import ast
 import dataclasses
 import json
-import pathlib
-from typing import Dict, List, Optional
 
-
-# Same as torchbenchmark.util.experiment.instantiator.TorchBenchModelConfig
-# https://github.com/pytorch/benchmark/blob/main/torchbenchmark/util/experiment/instantiator.py#L26
-@dataclasses.dataclass
-class TorchBenchModelConfig:
-    name: str
-    test: str
-    device: str
-    batch_size: Optional[int]
-    extra_args: List[str]
-    extra_env: Optional[Dict[str, str]] = None
-    output_dir: Optional[pathlib.Path] = None
+from typing import List
+from torchbenchmark.util.experiment.instantiator import TorchBenchModelConfig
 
 
 @dataclasses.dataclass