Skip to content

Commit c629fdc

Browse files
committed
Enhance Cutlass build process in Makefile with submodule initialization and CUDA architecture settings
1 parent 3e8b548 commit c629fdc

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/Makefile

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -429,6 +429,14 @@ clean_accelwattch_ubench:
429429
# cp cuda/cudnn/mnist/mnistCUDNN $(BINDIR)/$(BINSUBDIR)/
430430

431431
cutlass:
432+
mkdir -p $(BINDIR)/$(BINSUBDIR)/
433+
git submodule init && git submodule update
434+
mkdir -p cuda/cutlass/build
435+
$(SETENV) cmake -S cuda/cutlass -B cuda/cutlass/build -DCUTLASS_NVCC_ARCHS="80;90;90a" -DCMAKE_CUDA_ARCHITECTURES="80;90;90a" -DCMAKE_CUDA_COMPILER=$(CUDA_INSTALL_PATH)/bin/nvcc
436+
$(SETENV) $(MAKE) -C cuda/cutlass-bench/build/ cutlass_examples cutlass_profiler
437+
find cuda/cutlass-bench/build/examples -type f -executable -exec sh -c 'cp "$$1" "$(BINDIR)/$(BINSUBDIR)/cutlass_$$(basename "$$1")"' sh {} \;
438+
439+
cutlass_mini:
432440
mkdir -p $(BINDIR)/$(BINSUBDIR)/
433441
git submodule init && git submodule update
434442
mkdir -p cuda/cutlass-bench/build
@@ -441,6 +449,8 @@ cutlass:
441449

442450

443451

452+
453+
444454
cutlass_examples_turing:
445455
mkdir -p $(BINDIR)/$(BINSUBDIR)/
446456
git submodule init && git submodule update

0 commit comments

Comments
 (0)