From eb62433da1b3ab3429650faa244744120db9dcbd Mon Sep 17 00:00:00 2001
From: hlm628 <henry@hlm628.com>
Date: Sat, 23 Mar 2024 00:01:03 +0000
Subject: [PATCH] run on gpu

---
 Makefile | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/Makefile b/Makefile
index 167d071585b..21e128054ae 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,7 @@
-GPU=0
-CUDNN=0
-CUDNN_HALF=0
-OPENCV=0
+GPU=1
+CUDNN=1
+CUDNN_HALF=1
+OPENCV=1
 AVX=0
 OPENMP=0
 LIBSO=0
@@ -57,7 +57,7 @@ OS := $(shell uname)
 # ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]
 
 # Tesla A100 (GA100), DGX-A100, A30, A100, RTX 3080
-# ARCH= -gencode arch=compute_80,code=[sm_80,compute_80]
+ARCH= -gencode arch=compute_80,code=[sm_80,compute_80]
 
 # GeForce RTX 30x0, 30x0 Ti, Tesla GA10x, RTX Axxxx, A2, A10, A16, A40
 # ARCH= -gencode arch=compute_86,code=[sm_86,compute_86]