mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
0189ab6ab6
* docs + circle min size build * jit scan, arange, softmax * add sort * jit reductions * remove print * fix deps * clean includes / nits
10 lines
271 B
C++
10 lines
271 B
C++
// Copyright © 2024 Apple Inc.
|
|
|
|
constexpr std::string_view arange_kernels = R"(
|
|
template [[host_name("{0}")]] [[kernel]] void arange<{1}>(
|
|
constant const {1}& start,
|
|
constant const {1}& step,
|
|
device {1}* out,
|
|
uint index [[thread_position_in_grid]]);
|
|
)";
|