mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
@@ -16,6 +16,7 @@ MTL::ComputePipelineState* get_unary_kernel(
|
||||
metal::Device& d,
|
||||
const std::string& kernel_name,
|
||||
Dtype,
|
||||
Dtype,
|
||||
const std::string) {
|
||||
return d.get_kernel(kernel_name);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user