mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-29 21:11:16 +08:00
14 lines
234 B
Plaintext
14 lines
234 B
Plaintext
// Copyright © 2025 Apple Inc.
|
|
#pragma once
|
|
|
|
namespace mlx::core::cu {
|
|
|
|
struct Select {
|
|
template <typename T>
|
|
__device__ T operator()(bool condition, T x, T y) {
|
|
return condition ? x : y;
|
|
}
|
|
};
|
|
|
|
} // namespace mlx::core::cu
|