Skip to content

Commit

Permalink
fix multi gpu build
Browse files Browse the repository at this point in the history
  • Loading branch information
brandon-b-miller committed Aug 5, 2024
1 parent b4f3d94 commit cd5f19e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test_binary_generation/Makefile
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Generates the input files used by the pynvjitlink binding test suite

# Test binaries are built taking into account the CC of the GPU in the test machine
GPU_CC := $(shell nvidia-smi --query-gpu=compute_cap --format=csv | grep -v compute_cap | sed 's/\.//')
GPU_CC := $(shell nvidia-smi --query-gpu=compute_cap --format=csv | grep -v compute_cap | head -n 1 | sed 's/\.//')
GPU_CC ?= 75

# Use CC 7.0 as an alternative in fatbin testing, unless CC is 7.x
Expand Down

0 comments on commit cd5f19e

Please sign in to comment.