Rename cudnn_att.cu to cudnn_att.cpp to speed up compilation.

This commit is contained in:
Anerudhan 2024-05-08 19:53:48 +00:00
parent 1c5ef8cea3
commit 0ce5fcf7e8
2 changed files with 2 additions and 2 deletions

View file

@ -237,7 +237,7 @@ train_gpt2: train_gpt2.c
test_gpt2: test_gpt2.c
$(CC) $(CFLAGS) $(INCLUDES) $(LDFLAGS) $^ $(LDLIBS) $(OUTPUT_FILE)
$(NVCC_CUDNN): cudnn_att.cu
$(NVCC_CUDNN): cudnn_att.cpp
$(NVCC) -c $(NVCC_FLAGS) $(PFLAGS) $^ $(NVCC_INCLUDES)
train_gpt2cu: train_gpt2.cu $(NVCC_CUDNN)
@ -256,4 +256,4 @@ profile_gpt2cu: profile_gpt2.cu $(NVCC_CUDNN)
$(NVCC) $(NVCC_FLAGS) $(PFLAGS) -lineinfo $^ $(NVCC_LDFLAGS) $(NVCC_INCLUDES) $(NVCC_LDLIBS) $(CUDA_OUTPUT_FILE)
clean:
$(REMOVE_FILES) $(TARGETS)
$(REMOVE_FILES) $(TARGETS) $(NVCC_CUDNN)