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 Packages Projects Releases Wiki Activity

tmp

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-08-15 14:08:03 -07:00
parent 3b94e37270
commit 055c1ca929
4 changed files with 33 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/src/fast.cpp
View File

@@ -388,7 +388,7 @@ void init_fast(nb::module_& parent_module) {
m.def(
"precompiled_custom_kernel",
[](const std::string& name,
const std::string& compiled_source,
const nb::bytes compiled_source,
const std::vector<ScalarOrArray>& inputs_,
const std::vector<mx::Shape>& output_shapes,
const std::vector<mx::Dtype>& output_dtypes,
@@ -429,7 +429,9 @@ void init_fast(nb::module_& parent_module) {
return mx::fast::precompiled_custom_kernel(
name,
compiled_source,
std::string(
static_cast<const char*>(compiled_source.data()),
compiled_source.size()),
inputs,
output_shapes,
output_dtypes,