rocm

baptistecolle · baptistecolle · commit 6e16e18a9235 · 2024-10-29T08:45:20.000Z
diff --git a/.github/workflows/benchmark_rocm_pytorch.yaml b/.github/workflows/benchmark_rocm_pytorch.yaml
@@ -33,12 +33,12 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        subset: [unquantized, bnb, awq, gptq]
+        subset: [unquantized]
+        # [unquantized, bnb, awq, gptq]
 
         machine:
           [
-            { name: 1xA10, runs-on: { group: "aws-g5-4xlarge-plus" } },
-            { name: 1xT4, runs-on: { group: "aws-g4dn-2xlarge" } },
+            { name: 1xMI250, runs-on: [self-hosted, amd-gpu, single-gpu] },
           ]
 
     runs-on: ${{ matrix.machine.runs-on }}
diff --git a/.gitignore b/.gitignore
@@ -189,3 +189,6 @@ wip/
 *.csv
 optimum-benchmark/
 pyrsmi/
+
+requirements-dev.lock
+requirements.lock

Original file line number	Diff line number	Diff line change
`@@ -33,12 +33,12 @@ jobs:`
`33`	`33`	`strategy:`
`34`	`34`	`fail-fast: false`
`35`	`35`	`matrix:`
`36`		`- subset: [unquantized, bnb, awq, gptq]`
	`36`	`+ subset: [unquantized]`
	`37`	`+ # [unquantized, bnb, awq, gptq]`
`37`	`38`
`38`	`39`	`machine:`
`39`	`40`	`[`
`40`		`- { name: 1xA10, runs-on: { group: "aws-g5-4xlarge-plus" } },`
`41`		`- { name: 1xT4, runs-on: { group: "aws-g4dn-2xlarge" } },`
	`41`	`+ { name: 1xMI250, runs-on: [self-hosted, amd-gpu, single-gpu] },`
`42`	`42`	`]`
`43`	`43`
`44`	`44`	`runs-on: ${{ matrix.machine.runs-on }}`