diff --git a/examples/alexnet/CMakeLists.txt b/examples/alexnet/CMakeLists.txt index e919122..11a5b13 100644 --- a/examples/alexnet/CMakeLists.txt +++ b/examples/alexnet/CMakeLists.txt @@ -4,7 +4,7 @@ project(AlexNet LANGUAGES CXX CUDA ) -add_executable(alexnet main.cpp) +add_executable(alexnet alexnet.cpp) find_library(CUDANet_LIBRARY NAMES CUDANet HINTS ${CMAKE_CURRENT_SOURCE_DIR}/../../build) find_package(OpenCV REQUIRED COMPONENTS core imgcodecs imgproc) diff --git a/examples/alexnet/main.cpp b/examples/alexnet/alexnet.cpp similarity index 97% rename from examples/alexnet/main.cpp rename to examples/alexnet/alexnet.cpp index 9f31860..9edfdb2 100644 --- a/examples/alexnet/main.cpp +++ b/examples/alexnet/alexnet.cpp @@ -1,9 +1,4 @@ -#include -#include -#include -#include -#include -#include +#include #include #include #include