Test cpu

pytorch-labs · Dec 19, 2024 · f850878 · f850878
1 parent ba9e0d2
commit f850878
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -44,6 +44,8 @@ jobs:
           branch_name=${{ github.head_ref || github.ref_name }}
           # show disk space
           df -h
+          # show cpu
+          lscpu
           docker build . --memory 80g --shm-size 4g --build-arg TRITONBENCH_BRANCH="${branch_name}" --build-arg FORCE_DATE="${NIGHTLY_DATE}" \
               -f tritonbench-nightly.dockerfile -t ghcr.io/pytorch-labs/tritonbench:latest
           # Extract pytorch version from the docker