Packages

exla

0.7.2
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
5 Dependencies
Dependants
Files
Activity
exla c_src exla mlir custom_calls.h
Raw

c_src/exla/mlir/custom_calls.h 

#pragma once



template <typename DataType>

void qr_cpu_custom_call(void *out[], const void *in[]);



void qr_cpu_custom_call_bf16(void *out[], const void *in[]);

void qr_cpu_custom_call_f16(void *out[], const void *in[]);

void qr_cpu_custom_call_f32(void *out[], const void *in[]);

void qr_cpu_custom_call_f64(void *out[], const void *in[]);