View source on GitHub |
Functions
disable_custom_kernel(...)
: Prefer pure python operations to custom C++/CUDA kernel.
enable_custom_kernel(...)
: Prefer custom C++/CUDA kernel to pure python operations.
is_custom_kernel_disabled(...)
: Return whether custom C++/CUDA kernel is disabled.