zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add Primitive::name and remove Primitive::print (#2365)

Browse Source
This commit is contained in:
Cheng
2025-07-15 06:06:35 +09:00
committed by GitHub
parent 5201df5030
commit d34f887abc
32 changed files with 307 additions and 340 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/backend/metal/ternary.cpp
View File

@@ -11,7 +11,7 @@ namespace mlx::core {
void ternary_op_gpu_inplace(
const std::vector<array>& inputs,
array& out,
const std::string op,
const char* op,
const Stream& s) {
assert(inputs.size() == 3);
auto& a = inputs[0];
@@ -128,7 +128,7 @@ void ternary_op_gpu_inplace(
void ternary_op_gpu(
const std::vector<array>& inputs,
array& out,
const std::string op,
const char* op,
const Stream& s) {
auto& a = inputs[0];
auto& b = inputs[1];
@@ -141,13 +141,13 @@ void ternary_op_gpu(
void ternary_op_gpu(
const std::vector<array>& inputs,
array& out,
const std::string op) {
const char* op) {
auto& s = out.primitive().stream();
ternary_op_gpu(inputs, out, op, s);
}
void Select::eval_gpu(const std::vector<array>& inputs, array& out) {
ternary_op_gpu(inputs, out, get_primitive_string(this));
ternary_op_gpu(inputs, out, name());
}
} // namespace mlx::core