Skip to content

Commit

Permalink
Update CMakeLists.txt for required packages and optimize compiler fla…
Browse files Browse the repository at this point in the history
…gs; add debug output in normalise_vector function
  • Loading branch information
absternator committed Feb 25, 2025
1 parent c47a509 commit 2037dc7
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 4 deletions.
6 changes: 3 additions & 3 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -23,19 +23,19 @@ if(UNIX AND NOT APPLE)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D__STDC_LIMIT_MACROS -D__STDC_CONSTANT_MACROS")
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,--as-needed")
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -ffast-math -funroll-loops -m64")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -O1 -ffast-math -funroll-loops -m64")

# Set paths for non standard lib/ and include/ locations
if(DEFINED ENV{CONDA_PREFIX})
include_directories($ENV{CONDA_PREFIX}/include)
link_directories($ENV{CONDA_PREFIX}/lib)
link_directories($ENV{CONDA_PREFIX}/lib/intel64)
else()
find_package(OpenMP)
find_package(OpenMP REQUIRED) # Ensures OpenMP is found
endif()

# Add libraries
find_package(ZLIB)
find_package(ZLIB REQUIRED)
find_package(Python COMPONENTS Interpreter Development REQUIRED)
find_package(pybind11 CONFIG REQUIRED)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/vendor)
Expand Down
3 changes: 3 additions & 0 deletions src/vector_norm.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,11 @@ inline void normalise_vector(std::vector<T> &vec, const bool check_positive,
#pragma omp parallel for schedule(static) reduction(+: sum) reduction(&: all_positive) num_threads(n_threads)
for (uint64_t it = 0; it < vec.size(); ++it) {
sum += vec[it];
std::cout << vec[it] << std::endl;
all_positive &= vec[it] >= 0;
}
std::cout << "Positive: " << all_positive << std::endl;
std::cout << "Sum: " << sum << std::endl;

if (check_positive && !all_positive) {
throw std::runtime_error("Probability vector has negative entries");
Expand Down
1 change: 0 additions & 1 deletion src/wtsne.hpp
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
#pragma once

#include <wchar.h>
#include <algorithm>
#include <array>
#include <chrono>
Expand Down

0 comments on commit 2037dc7

Please sign in to comment.