JuliaGPU / JuliaGPU/Metal.jl

EnzymeRules for MTLArray Constructor

Open
#414 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Julia
Stars
463
Forks
68
Avg merge
1d 19m
Merged PRs (30d)
32

Description

See https://github.com/EnzymeAD/Enzyme.jl/issues/1771 (but we now also ought do for MTL)

CUDA.jl has Enzyme rules for CuArray constructors, and kernel launches. See https://github.com/JuliaGPU/CUDA.jl/blob/master/ext/EnzymeCoreExt.jl

Importing MTLArray versions of these here would enable a similar level of functionality on Metal and likely benefit many users.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.