GPU := H100

SRC := linear_attn.cu
OUT := linear_attn$(shell python3 -c "import sysconfig; print(sysconfig.get_config_var('EXT_SUFFIX'))")
CMD := python bench.py
CONFIG := python

# Additional flags not in common.mk (include path comes from common.mk)
NVCCFLAGS += -lcublas

include ../common.mk
