From 17114558ec5c24995158d14d8ec6780fd95eab96 Mon Sep 17 00:00:00 2001 From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com> Date: Tue, 8 Oct 2024 23:41:28 +0530 Subject: [PATCH] removed cuda device --- .github/workflows/test-mlperf-inference-mixtral.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test-mlperf-inference-mixtral.yml b/.github/workflows/test-mlperf-inference-mixtral.yml index d7a358ae13..b5716e78f8 100644 --- a/.github/workflows/test-mlperf-inference-mixtral.yml +++ b/.github/workflows/test-mlperf-inference-mixtral.yml @@ -16,7 +16,7 @@ jobs: matrix: python-version: [ "3.12" ] backend: [ "pytorch" ] - device: [ "cpu", "cuda" ] + device: [ "cpu" ] steps: - name: Test MLPerf Inference MIXTRAL-8X7B reference implementation