jax.extend.ffi 模块

jax.extend.ffi 模块#

ffi_call(target_name, result_shape_dtypes, *args)

调用外部函数接口 (FFI) 目标。

ffi_lowering(call_target_name, *[, ...])

为外部函数接口 (FFI) 目标构建降低规则。

pycapsule(funcptr)

将 ctypes 函数指针包装在 PyCapsule 中。

register_ffi_target(name, fn[, platform, ...])

注册外部函数目标。