Revert "reorg"

This reverts commit 03624d0.
avijaira · Nov 13, 2024 · 418f931 · 418f931
1 parent 03624d0
commit 418f931
Show file tree

Hide file tree

Showing 21 changed files with 763 additions and 480 deletions.
diff --git a/kernels/attn/4090_harness.impl b/kernels/attn/4090_harness.impl
diff --git a/kernels/attn/Makefile b/kernels/attn/Makefile
diff --git a/kernels/attn/a100.cu b/kernels/attn/a100.cu
diff --git a/kernels/attn/4090.cu → kernels/attn/demo/4090.cu b/kernels/attn/4090.cu → kernels/attn/demo/4090.cu
diff --git a/kernels/attn/a100_harness.impl → kernels/attn/demo/4090_harness.impl b/kernels/attn/a100_harness.impl → kernels/attn/demo/4090_harness.impl
diff --git a/kernels/attn/demo/Makefile b/kernels/attn/demo/Makefile
@@ -0,0 +1,24 @@
+# Compiler
+NVCC=nvcc
+GPU=4090
+
+# Conditional setup based on the target GPU
+ifeq ($(GPU),4090)
+NVCCFLAGS=-DNDEBUG -Xcompiler=-fPIE --expt-extended-lambda --expt-relaxed-constexpr -Xcompiler=-Wno-psabi -Xcompiler=-fno-strict-aliasing --use_fast_math -I../../../include -forward-unknown-to-host-compiler -O3 -Xnvlink=--verbose -Xptxas=--verbose -Xptxas=--warn-on-spills -std=c++20 -MD -MT -MF -x cu -lrt -lpthread -ldl -DKITTENS_4090 -arch=sm_89 -lcuda -lcudadevrt -lcudart_static -lcublas
+TARGET=attn_fwd
+SRC=4090.cu
+else
+NVCCFLAGS=-DNDEBUG -Xcompiler=-fPIE --expt-extended-lambda --expt-relaxed-constexpr -Xcompiler=-Wno-psabi -Xcompiler=-fno-strict-aliasing --use_fast_math -I../../../include -forward-unknown-to-host-compiler -O3 -Xnvlink=--verbose -Xptxas=--verbose -Xptxas=--warn-on-spills -std=c++20 -MD -MT -MF -x cu -lrt -lpthread -ldl -DKITTENS_HOPPER -arch=sm_90a -lcuda -lcudadevrt -lcudart_static -lcublas
+TARGET=attn_fwd
+SRC=h100_lcf.cu
+endif
+
+# Default target
+all: $(TARGET)
+
+$(TARGET): $(SRC)
+	$(NVCC) $(SRC) $(NVCCFLAGS) -o $(TARGET)
+
+# Clean target
+clean:
+	rm -f $(TARGET)
diff --git a/kernels/attn/lcf_demo/README.md → kernels/attn/demo/README.md b/kernels/attn/lcf_demo/README.md → kernels/attn/demo/README.md