Update metadata.yaml

pytorch · Jul 20, 2023 · 6612b8e · 6612b8e
1 parent b955230
commit 6612b8e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torchbenchmark/models/llama_v2/metadata.yaml b/torchbenchmark/models/llama_v2/metadata.yaml
@@ -6,5 +6,11 @@ eval_deterministic: false
 eval_nograd: true
 not_implemented:
 - jit: true
+- device: cpu
+  test: train
+- device: cuda
+  test: eval
+- device: cuda
+  test: train
 train_benchmark: false
 train_deterministic: false