publishing

Oneflow-Inc · Dec 22, 2023 · 646287d · 646287d
1 parent 5f030f9
commit 646287d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/utils/matrix.ts b/src/utils/matrix.ts
@@ -52,13 +52,13 @@ function getRunsOn(test: Test, deviceLabel: Device): string[] {
   return runnerLabels.concat([`${deviceLabel}-${suffix}`])
 }
 
-export type ComputePlatform = 'cpu' | 'cu116' | 'cu110_xla' | 'cu101_xla'
+export type ComputePlatform = 'cpu' | 'cu118' | 'cu110_xla' | 'cu101_xla'
 function getComputePlatform(device: Device): ComputePlatform {
   switch (device) {
     case 'cpu':
       return 'cpu'
     case 'cuda':
-      return 'cu116'
+      return 'cu118'
 
     default:
       throw new Error(device)