Manage runtime environments (#452)

* Add `Env` class that manages all runtime environments.
* Changed `NPKIT_DUMP_DIR` to `MSCCLPP_NPKIT_DUMP_DIR`.
This commit is contained in:
Changho Hwang
2025-01-15 09:44:52 -08:00
committed by GitHub
parent 8ac50dc85d
commit 869cdba00c
19 changed files with 229 additions and 51 deletions

View File

@@ -4,6 +4,7 @@
import os as _os
from ._mscclpp import (
Env,
ErrorCode,
BaseError,
Error,
@@ -32,6 +33,7 @@ from ._mscclpp import (
ExecutionPlan,
PacketType,
RawGpuBuffer,
env,
version,
is_nvls_supported,
npkit,