| 435 | } |
| 436 | |
| 437 | CUfunction JitModule::get_kernel( |
| 438 | const std::string& kernel_name, |
| 439 | std::function<void(CUfunction)> configure_kernel) { |
| 440 | return get_kernel_and_dims(kernel_name, std::move(configure_kernel)).first; |
| 441 | } |
| 442 | |
| 443 | JitModule& get_jit_module( |
| 444 | const mlx::core::Device& device, |
no outgoing calls
no test coverage detected