The nvmath.bindings module is experimental and potentially subject to future changes.

set_workspace#

nvmath.bindings.cublas.set_workspace(
intptr_t handle,
intptr_t workspace,
size_t workspace_size_in_bytes,
)[source]#

See cublasSetWorkspace.