zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-30 06:31:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

cpu: allow complex logaddexp / atan2

Browse Source
This commit is contained in:
Yury Popov 2025-04-20 02:06:28 +03:00
parent c7e326bb44
commit e444e2cbc0
No known key found for this signature in database
GPG Key ID: 76DE18AD6634F257
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/backend/cpu/binary.cpp
View File

@ -172,9 +172,12 @@ void binary_float(
case bfloat16: case bfloat16:
binary_op<bfloat16_t, Op>(a, b, out, bopt); binary_op<bfloat16_t, Op>(a, b, out, bopt);
break; break;
case complex64:
binary_op<complex64_t, Op>(a, b, out, bopt);
break;
default: default:
throw std::runtime_error( throw std::runtime_error(
"[binary_float] Only supports non-complex floating point types."); "[binary_float] Only supports floating point types.");
} }
}); });
} }