summaryrefslogtreecommitdiff
path: root/cuda-kernels/Makefile
diff options
context:
space:
mode:
Diffstat (limited to 'cuda-kernels/Makefile')
-rwxr-xr-xcuda-kernels/Makefile3
1 files changed, 2 insertions, 1 deletions
diff --git a/cuda-kernels/Makefile b/cuda-kernels/Makefile
index 51a7760..673460f 100755
--- a/cuda-kernels/Makefile
+++ b/cuda-kernels/Makefile
@@ -1,5 +1,6 @@
all: tensor_core.cu
- nvcc -arch=sm_70 -lcudart -g -o tensor_core tensor_core.cu
+ nvcc --gpu-architecture=compute_70 --gpu-code=compute_70 -lcudart -g -o tensor_core tensor_core.cu
+# nvcc -arch=sm_70 -lcudart -g -o tensor_core tensor_core.cu
.PHONY:
clean: