From bb9c895400e2771dd9886e8dc0443ab89f4ff56b Mon Sep 17 00:00:00 2001 From: Erik Schultheis Date: Thu, 6 Jun 2024 12:02:33 +0300 Subject: [PATCH] fix --- .github/workflows/ci.yml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index e10adbe..27ebad6 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -152,8 +152,10 @@ jobs: nvcc --version g++ --version - - name: Get CUDNN - run: git clone https://github.com/NVIDIA/cudnn-frontend.git + - name: Install cudnn frontend + run: | + apt-get update && apt-get install -y git + git clone https://github.com/NVIDIA/cudnn-frontend.git - name: Build FP32 checkpoint run: make train_gpt2fp32cu test_gpt2fp32cu