compile_matmul_epilog#

nvmath.sparse.compile_matmul_epilog(epilog, *, dtype, compute_capability=None)[source]#

Compile a unary Python function with argument type dtype returning a value of type dtype to device code.

Parameters:

  • epilog – The unary Python function to compile to device code.

  • dtype – The dtype of the epilog’s argument. Any scalar Numba type can be provided.

  • compute_capability – The target compute capability, specified as a string ('80', '89', …). The default is the compute capability of the current device.

Returns:

The compiled device code as a bytes object.