Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 9 additions & 7 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,6 @@ CONDA_ENV_NAME := vortex

PYTHON_VERSION := $(shell $(PYTHON) -c 'import sys; print(sys.version_info[0])')
TARGET_PYTHON_VERSION := 3.11
CUDA_PATH := /usr/local/cuda
CUDA_INCLUDE_PATH := $(CUDA_PATH)/include
CUDA_LIB_PATH := $(CUDA_PATH)/lib64
CPATH := $(CUDA_INCLUDE_PATH):/usr/local/cuda/include
CUDACXX := /usr/local/cuda/bin/nvcc
CUDA_HOME := $(CUDA_PATH)

_detect_cuda_path:
ifndef CUDA_PATH
Expand All @@ -21,9 +15,17 @@ ifndef CUDA_PATH
# Derive CUDA_PATH by going one directory up from nvcc's bin directory
CUDA_PATH := $(dir $(NVCC_BIN))..
CUDA_PATH := $(realpath $(CUDA_PATH))
else
CUDA_PATH := /usr/local/cuda
endif
endif

CUDA_INCLUDE_PATH := $(CUDA_PATH)/include
CUDA_LIB_PATH := $(CUDA_PATH)/lib64
CPATH := $(CUDA_INCLUDE_PATH):/usr/local/cuda/include
CUDACXX := $(CUDA_PATH)/bin/nvcc
CUDA_HOME ?= $(CUDA_PATH)

_check_env_enabled:
ifneq ($(VIRTUAL_ENV),)
$(info Detected active Python venv at $(VIRTUAL_ENV))
Expand Down Expand Up @@ -88,4 +90,4 @@ format:
ruff check --fix .
black -l 120 .

rebuild: clean setup
rebuild: clean setup-full