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]]);
)"