nvalchemiops.math and nvalchemiops.types: Utilities

Mathematical Utilities

nvalchemiops.math.wpdivmod(a, b)

Warp implementation of the divmod utility.

Parameters:

• a (int)

• b (int)

Type Conversion Utilities

nvalchemiops.types.get_wp_dtype(dtype)

Get the warp dtype for a given torch dtype.

Parameters:

dtype (dtype)

nvalchemiops.types.get_wp_mat_dtype(dtype)

Get the warp mat dtype for a given torch dtype.

Parameters:

dtype (dtype)

nvalchemiops.types.get_wp_vec_dtype(dtype)

Get the warp vec dtype for a given torch dtype.

Parameters:

dtype (dtype)

nvalchemiops.types.get_wp_dtype(dtype)

Get the warp dtype for a given torch dtype.

Parameters:

dtype (dtype)

nvalchemiops.types.get_wp_vec_dtype(dtype)

Get the warp vec dtype for a given torch dtype.

Parameters:

dtype (dtype)

nvalchemiops.types.get_wp_mat_dtype(dtype)

Get the warp mat dtype for a given torch dtype.

Parameters:

dtype (dtype)