Skip to content

Commit b8eaafd

Browse files
authored
Merge pull request karpathy#386 from Anerudhan/feature/rename_cudnn_attn
Rename cudnn_att.cu to cudnn_att.cpp to speed up compilation.
2 parents 68509c8 + 0ce5fcf commit b8eaafd

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Makefile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ train_gpt2: train_gpt2.c
237237
test_gpt2: test_gpt2.c
238238
$(CC) $(CFLAGS) $(INCLUDES) $(LDFLAGS) $^ $(LDLIBS) $(OUTPUT_FILE)
239239

240-
$(NVCC_CUDNN): cudnn_att.cu
240+
$(NVCC_CUDNN): cudnn_att.cpp
241241
$(NVCC) -c $(NVCC_FLAGS) $(PFLAGS) $^ $(NVCC_INCLUDES)
242242

243243
train_gpt2cu: train_gpt2.cu $(NVCC_CUDNN)
@@ -256,4 +256,4 @@ profile_gpt2cu: profile_gpt2.cu $(NVCC_CUDNN)
256256
$(NVCC) $(NVCC_FLAGS) $(PFLAGS) -lineinfo $^ $(NVCC_LDFLAGS) $(NVCC_INCLUDES) $(NVCC_LDLIBS) $(CUDA_OUTPUT_FILE)
257257

258258
clean:
259-
$(REMOVE_FILES) $(TARGETS)
259+
$(REMOVE_FILES) $(TARGETS) $(NVCC_CUDNN)
File renamed without changes.

0 commit comments

Comments
 (0)