Manx (manx v0.1.4)
Manx is a MLIR backend for the Nx. It mainly targets TOSA/Linalg dialect and will generate LLVM/CUDA/Vulkan code for different configurations.
Link to this section Summary
Functions
Add returned memref to the allocator.
Create a new tensor of null ptr memref. This should be used as as the return tensor of JIT function.
Link to this section Functions
Link to this function
add_allocated_memory(tensor)
Add returned memref to the allocator.
Link to this function
jit(function, args, options \\ [])
Link to this function
tensor_of_null_memref(tensor)
Create a new tensor of null ptr memref. This should be used as as the return tensor of JIT function.