From ef3a62f399185e52f36fd4adf61a261a15a89b49 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Thu, 18 Apr 2024 20:44:32 +0200
Subject: [PATCH] Remove hardcoded cuda architecture

---
 CMakeLists.txt                | 1 -
 README.md                     | 2 +-
 tools/export_model_weights.py | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index bfcd083..698edbb 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -19,7 +19,6 @@ set(LIBRARY_SOURCES
     ${LIBRARY_SOURCES}
 )
 
-set(CMAKE_CUDA_ARCHITECTURES 75)
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 
 # Build static library
diff --git a/README.md b/README.md
index abc7c56..a3d178e 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,7 @@ Convolutional Neural Network inference library running on CUDA.
 ```sh
 mkdir build
 cd build
-cmake -S ..
+cmake -S .. -DCMAKE_CUDA_ARCHITECTURES=75  # Replace with you cuda architecture
 make
 ```
 
diff --git a/tools/export_model_weights.py b/tools/export_model_weights.py
index e9f839e..4eb776a 100644
--- a/tools/export_model_weights.py
+++ b/tools/export_model_weights.py
@@ -25,4 +25,4 @@ def export_model_weights(model: torch.nn.Module, filename):
 
         f.seek(0)
         f.write(struct.pack('q', len(header)))            
-        f.write(header.encode('utf-8'))
\ No newline at end of file
+        f.write(header.encode('utf-8'))