Packages

exla

0.12.0
0.13.0 0.12.0 0.11.0 0.10.0 0.9.2 0.9.1 0.9.0 0.8.0 0.7.3 0.7.2 0.7.1 0.7.0 0.6.4 0.6.3 0.6.1 0.6.0 0.5.3 0.5.2 0.5.1 0.5.0 0.4.2 0.4.1 0.4.0 0.3.0 0.2.3 0.2.2 0.2.1 0.2.0

Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx

HexDocs

Current section

Files

Jump to
Readme
28 Versions
6 Dependencies
Dependants
Files
Activity
exla c_src exla_test custom_calls.cc
Raw

c_src/exla_test/custom_calls.cc 

// Test-only shared library: registers an alias FFI name that reuses the

// existing qr_cpu_custom_call_f32 handler symbol from libxla_extension.so.

#ifndef EXLA_PROD



#include "xla/ffi/api/api.h"

#include "xla/ffi/ffi_api.h"



namespace ffi = xla::ffi;



extern "C" XLA_FFI_Error *qr_cpu_custom_call_f32(XLA_FFI_CallFrame *call_frame);



XLA_FFI_REGISTER_HANDLER(ffi::GetXlaFfiApi(), "qr_cpu_custom_call_f32_exla_alias",

                         "Host", qr_cpu_custom_call_f32);



#endif