feat(week1): add docs and task for week 1.

This commit is contained in:
Yi Pan
2024-07-09 11:45:35 +08:00
parent f0a44579c4
commit 66430108a4
5 changed files with 256 additions and 87 deletions

32
Makefile Normal file
View File

@ -0,0 +1,32 @@
CC = nvcc
CFLAGS = -I/usr/local/lib/python3.10/dist-packages/torch/include \
-I/usr/local/lib/python3.10/dist-packages/torch/include/torch/csrc/api/include \
-lcublas
HELLO = build/hello
HELLO_SRC = csrc/hello_world.cu
BASIC = build/basic
BASIC_SRC = csrc/basic.cu
GEMM = build/gemm
GEMM_SRC = csrc/gemm.cu
all: $(HELLO) $(BASIC)
$(HELLO): $(HELLO_SRC)
mkdir -p build
$(CC) $(CFLAGS) $^ -o $@
$(BASIC): $(BASIC_SRC)
mkdir -p build
$(CC) $(CFLAGS) $^ -o $@
$(GEMM): $(GEMM_SRC)
mkdir -p build
$(CC) $(CFLAGS) $^ -o $@
clean:
rm -rf build
.PHONY: all clean