MLX
Loading...
Searching...
No Matches
Variables
unary.h File Reference

Go to the source code of this file.

Variables

constexpr std::string_view unary_kernels
 

Variable Documentation

◆ unary_kernels

constexpr std::string_view unary_kernels
constexpr
Initial value:
= R"(
template [[host_name("v{0}")]] [[kernel]] void unary_v<{1}, {2}>(
device const {1}* in,
device {1}* out,
uint index [[thread_position_in_grid]]);
template [[host_name("g{0}")]] [[kernel]] void unary_g<{1}, {2}>(
device const {1}* in,
device {1}* out,
device const int* in_shape,
device const size_t* in_strides,
device const int& ndim,
uint index [[thread_position_in_grid]]);
)"