Allow llama_v2_7b_16h to run on A100.

pytorch · Jan 22, 2024 · f7d6d9d · f7d6d9d
1 parent 8f30da3
commit f7d6d9d
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/torchbenchmark/models/llama_v2_7b_16h/metadata.yaml b/torchbenchmark/models/llama_v2_7b_16h/metadata.yaml
@@ -7,9 +7,5 @@ eval_nograd: true
 not_implemented:
 - device: cpu
 - device: NVIDIA A10G
-# TODO: llama_v2_7b_16h accuracy test will cause "CUBLAS_STATUS_NOT_INITIALIZED" Error
-# https://github.com/pytorch/benchmark/issues/2064
-- device: NVIDIA A100-SXM4-40GB
-  test: eval
 train_benchmark: false
 train_deterministic: false