manx v0.1.4

Manx.Compiler (manx v0.1.4)

Link to this section Summary

Functions

invoke(return, args, jit, symbol)

Invoke MLIR JIT with Nx tensors. If there are tuples their memrefs will be packed into a single C struct.

memref_from_tensor(f)
  • If it is a tensor, return a memref
  • If it is a tuple, recursively pack them into one struct.
populate_tensor_from_memref(tensor, memory)
  • If it is a tensor, return a memref
  • If it is a tuple, recursively unpack each member from the nested struct.

Link to this section Functions

Link to this function

invoke(return, args, jit, symbol)

Invoke MLIR JIT with Nx tensors. If there are tuples their memrefs will be packed into a single C struct.

Link to this function

memref_from_tensor(f)

  • If it is a tensor, return a memref
  • If it is a tuple, recursively pack them into one struct.
Link to this function

populate_tensor_from_memref(tensor, memory)

  • If it is a tensor, return a memref
  • If it is a tuple, recursively unpack each member from the nested struct.