Skip to content

Affine registration on the GPU #3258

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
daljit46 wants to merge 12 commits into
base: dev
Choose a base branch
from
Draft

Affine registration on the GPU #3258

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
0102320
Add magic_enum dependency
daljit46 Jan 14, 2026
73e1189
First draft of affine registration on GPU
daljit46 Jan 14, 2026
022149f
Symlink shader code + hack to find registration shaders
daljit46 Jan 14, 2026
9384096
Use exec path for finding shaders
daljit46 Jan 14, 2026
0f76ded
Export boolean link-time constants
daljit46 Jan 14, 2026
8866f5f
Update registration uniforms for new GPU API
daljit46 Jan 19, 2026
cff396b
Add write_object_to_buffer helper
daljit46 Jan 19, 2026
983f1f1
Add test for write_object_to_buffer
daljit46 Jan 19, 2026
b412e03
Enforce 4-byte size for GPU object uploads
daljit46 Jan 19, 2026
3bb7533
Rename Metric class -> GlobalMetric
daljit46 Jan 29, 2026
0a6b4e2
Rename RegistrationConfig -> GlobalRegistrationConfig
daljit46 Jan 29, 2026
cfb2dca
Add missing include header in registrationtypes.h
daljit46 Jan 29, 2026
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
1 change: 1 addition & 0 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ option(MRTRIX_USE_SYSTEM_NIFTI "Use system-installed NIfTI C headers" OFF)
option(MRTRIX_USE_SYSTEM_GTEST "Use system-installed Google Test library" OFF)
option(MRTRIX_USE_SYSTEM_DAWN "Use system-installed Dawn library" OFF)
option(MRTRIX_USE_SYSTEM_SLANG "Use system-installed Slang library" OFF)
option(MRTRIX_USE_SYSTEM_MAGIC_ENUM "Use system-installed Magic Enum library" OFF)
option(MRTRIX_USE_SYSTEM_TCB_SPAN "Use system-installed TCB Span library" OFF)

if(MRTRIX_BUILD_TESTS)
Expand Down
16 changes: 14 additions & 2 deletions cmake/Dependencies.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,11 +42,11 @@ endif()

# Nifti headers
if(MRTRIX_USE_SYSTEM_NIFTI)
find_path(NIFTI1_INCLUDE_DIR
find_path(NIFTI1_INCLUDE_DIR
NAMES nifti1.h
PATHS ${NIFTI_DIR} /usr/include /usr/local/include
)
find_path(NIFTI2_INCLUDE_DIR
find_path(NIFTI2_INCLUDE_DIR
NAMES nifti2.h
PATHS ${NIFTI_DIR} /usr/include /usr/local/include
)
Expand Down Expand Up @@ -236,3 +236,15 @@ else()
add_library(tcb::span ALIAS tcb_span)
endif()

# magic_enum
if(MRTRIX_USE_SYSTEM_MAGIC_ENUM)
find_package(magic_enum 0.9.7 CONFIG REQUIRED)
else()
set(magic_enum_url "https://github.com/Neargye/magic_enum/archive/refs/tags/v0.9.7.tar.gz")
FetchContent_Declare(
magic_enum
DOWNLOAD_EXTRACT_TIMESTAMP ON
URL ${magic_enum_url}
)
FetchContent_MakeAvailable(magic_enum)
endif()
Loading