emelryan's picture
Duplicate from nvidia/nemotron-ocr-v2-multilingual
b28505d
Raw
History Blame Contribute Delete
354 Bytes
ARCH ?= sm_70
LOWER_BOUND ?= 1
UPPER_BOUND ?= 16
CUDA_HOME ?= /usr/local/cuda
NVCC = $(CUDA_HOME)/bin/nvcc
all: benchmark block
benchmark: benchmark.cu
$(NVCC) -arch=$(ARCH) -I../ -Xptxas -v benchmark.cu -o benchmark -DLOWER_BOUND=$(LOWER_BOUND) -DUPPER_BOUND=$(UPPER_BOUND)
block: block.cu
$(NVCC) -arch=$(ARCH) -I../ -Xptxas -v block.cu -o block