Utils¶
Internal utilities used by WarpConvNet. These are not part of the public API and may change without notice.
| Module | Purpose |
|---|---|
warpconvnet.utils.benchmark_cache |
Persistent auto-tuning cache (msgpack) |
warpconvnet.utils.autotune_warmup |
Scripts for pre-populating the benchmark cache |
warpconvnet.utils.timer |
CUDA-aware timing helpers |
warpconvnet.utils.logger |
Logging setup |
warpconvnet.utils.dist |
Distributed training helpers |
warpconvnet.utils.type_cast |
Mixed-precision dtype utilities |
warpconvnet.utils.ravel |
N-dimensional index raveling/unraveling |
warpconvnet.utils.unique |
Unique-value operations on tensors |
warpconvnet.utils.argsort |
Argsort utilities |
warpconvnet.utils.nested |
Nested tensor helpers |
warpconvnet.utils.ntuple |
Tuple padding/expansion (like torch.nn.modules.utils) |