Skip to content

Mark CUDACore.cudaError_enum as having no tangent #1175

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

Merged
sunxd3 merged 2 commits into from
May 15, 2026
Merged

Mark CUDACore.cudaError_enum as having no tangent #1175

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6886e37
Update MooncakeCUDAExt.jl
kshyatt May 14, 2026
ce251a3
Apply Xianda Sun's suggestion
kshyatt May 15, 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
6 changes: 4 additions & 2 deletions ext/MooncakeCUDAExt/MooncakeCUDAExt.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ using CUDA.CUDACore:
is_capturing,
capture_status,
hasfieldcount
using CUDA: CUDACore
using CUDA: cuBLAS
using CUDA: cuSPARSE
using CUDA: cuSOLVER
Expand Down Expand Up @@ -313,13 +314,13 @@ _register_cuda_opaque_types!()
# CUDA @cenum types are primitive types (integer-backed C enums) — never differentiable.
# Mooncake's generic tangent_type @generated function errors on primitive types with no
# registered method, so we register all of them here programmatically.
# Covers: cuBLAS, cuSPARSE, cuSOLVER.
# Covers: CUDACore, cuBLAS, cuSPARSE, cuSOLVER.
# cuDNN enums are handled in MooncakeCUDNNExt (loaded only when cuDNN is available).
# Filter: parentmodule(T) must be one of the CUDA family modules, to avoid accidentally
# re-registering standard Julia primitive types (Bool, Int32, Float64, ...) that happen
# to be visible in the CUDA namespace.
function _register_cuda_enum_types!()
let _cuda_family = (cuBLAS, cuSPARSE, cuSOLVER)
let _cuda_family = (CUDACore, cuBLAS, cuSPARSE, cuSOLVER)
_cenum_seen = Set{DataType}()
for _mod in _cuda_family
for _nm in names(_mod; all=true)
Expand All @@ -330,6 +331,7 @@ function _register_cuda_enum_types!()
end
_T isa DataType || continue
isprimitivetype(_T) || continue
_T <: CUDACore.CEnum.Cenum || continue
parentmodule(_T) in _cuda_family || continue
_T in _cenum_seen && continue
push!(_cenum_seen, _T)
Expand Down
Loading