[FIX] Ensure tensors consistently move to backend device and add tests for GPU-specific behavior - #90
Merged
Merged
Conversation
Member
Author
This was referenced Jul 27, 2026
Merged
Merged
Leguark
force-pushed
the
keops_race_condition
branch
from
September 4, 2026 13:17
fed06f6 to
d2d0f6a
Compare
This was referenced Sep 4, 2026
Leguark
marked this pull request as ready for review
September 4, 2026 13:29
…s for GPU-specific behavior Aligns tensor device in `_repeat` and `_array` helpers to backend settings. Adds regression tests to validate PyTorch GPU device handling and tensor movement in preprocessing and repeated operations.
…evel handling Introduces thread-safe handling of mutable option states to prevent collisions during concurrent computations. Adds parameterized octree-level support in tests and refactors option initialization for request-level isolation. Includes validations for stacked kernel dimension mismatches and updates relevant symbolic evaluator logic.
…functions`, `main_server_pro`) and related dependencies Cleans up redundant code and unused FastAPI server utilities. Prepares the codebase for better maintainability and focuses on removing legacy server-side logic.
…diagonal matrix construction Replaces repetitive `arange` implementations with a unified method in `BackendTensor`. Refactors covariance assemblers to leverage the centralized logic for improved readability and maintainability.
Leguark
force-pushed
the
keops_race_condition
branch
from
September 5, 2026 09:12
18a2912 to
abea548
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
[FIX] Ensure tensors consistently move to backend device and add tests for GPU-specific behavior
Aligns tensor device in
_repeatand_arrayhelpers to backend settings. Adds regression tests to validate PyTorch GPU device handling and tensor movement in preprocessing and repeated operations.[ENH] Add concurrency safety for mutable options and enhance octree-level handling
Introduces thread-safe handling of mutable option states to prevent collisions during concurrent computations. Adds parameterized octree-level support in tests and refactors option initialization for request-level isolation. Includes validations for stacked kernel dimension mismatches and updates relevant symbolic evaluator logic.