update

Author: shink

.github/workflows/_pytorch_pr_comment.yml

@@ -38,7 +38,6 @@ jobs:
           issue-number: ${{ inputs.pr-number }}
           comment-author: ${{ inputs.pr-comment-author }}
           body-includes: '<!-- ACCELERATOR TESTINFRA START -->'
-          direction: first
 
       - name: Show comment info
         run: |

.github/workflows/ascend_npu_test.yml

@@ -119,6 +119,7 @@ jobs:
       - prepare
       - build
       - test
+    if: ${{ github.event_name == 'repository_dispatch' }}
     uses: ./.github/workflows/_ascend_npu_benchmark.yml
     with:
       runner: ${{ needs.prepare.outputs.runner }}

requirements.txt

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

src/benchmark/utils.py

@@ -1,9 +1,21 @@
 import ast
 import dataclasses
 import json
+import pathlib
+from typing import Dict, List, Optional
 
-from typing import List
-from torchbenchmark.util.experiment.instantiator import TorchBenchModelConfig
+
+# 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
 
 
 @dataclasses.dataclass