Skip to content

[gpu] Support AOT offloading of tensors#57

Merged
guillon merged 1 commit intoxtc-tools:mainfrom
ElectrikSpace:dev/snoiry/gpu_tensor_offload
Apr 13, 2026
Merged

[gpu] Support AOT offloading of tensors#57
guillon merged 1 commit intoxtc-tools:mainfrom
ElectrikSpace:dev/snoiry/gpu_tensor_offload

Conversation

@ElectrikSpace
Copy link
Copy Markdown
Contributor

Implement tensor offloading on the GPU target, using the MLIR Cuda runner runtime functions.

DO NOT MERGE FIRST
-> Only the last commit contains the feature, the others are dependencies from:

@ElectrikSpace ElectrikSpace self-assigned this Feb 24, 2026
@guillon guillon added the enhancement New feature or request label Feb 27, 2026
Comment thread src/xtc/runtimes/accelerator/gpu/__init__.py
Comment thread tests/filecheck/backends/target_gpu/test_matmul_mlir_offload_tensor.py Outdated
Comment thread src/xtc/targets/accelerator/gpu/GPUEvaluator.py
@ElectrikSpace ElectrikSpace force-pushed the dev/snoiry/gpu_tensor_offload branch 3 times, most recently from 450c6a2 to 8553e8b Compare March 11, 2026 14:32
@ElectrikSpace ElectrikSpace requested a review from guillon March 11, 2026 14:51
Add support for ahead of time copy of tensors on the GPU target, relying
on the Mlir runner runtime functions for GPUs.
@ElectrikSpace ElectrikSpace force-pushed the dev/snoiry/gpu_tensor_offload branch from 8553e8b to ca35787 Compare April 13, 2026 15:45
@guillon guillon merged commit 6cf9bb7 into xtc-tools:main Apr 13, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants