tilelang.env¶

Attributes¶

`logger`
`CUTLASS_NOT_FOUND_MESSAGE`	, which may lead to compilation bugs when utilize tilelang backend.
`COMPOSABLE_KERNEL_NOT_FOUND_MESSAGE`	, which may lead to compilation bugs when utilize tilelang backend.
`TL_TEMPLATE_NOT_FOUND_MESSAGE`	, which may lead to compilation bugs when utilize tilelang backend.
`TVM_LIBRARY_NOT_FOUND_MESSAGE`
`enable_cache`
`disable_cache`
`is_cache_enabled`
`env`
`CUDA_HOME`
`ROCM_HOME`
`install_tvm_path`
`install_cutlass_path`
`install_ck_path`
`install_tl_template_path`
`CUTLASS_INCLUDE_DIR`
`COMPOSABLE_KERNEL_INCLUDE_DIR`
`TILELANG_TEMPLATE_PATH`

Classes¶

`CacheState`	Class to manage global kernel caching state.
`EnvVar`	Descriptor for managing access to a single environment variable.
`Environment`	Environment configuration for TileLang.

Module Contents¶

tilelang.env.logger¶

tilelang.env.CUTLASS_NOT_FOUND_MESSAGE = 'CUTLASS is not installed or found in the expected path'¶: , which may lead to compilation bugs when utilize tilelang backend.

tilelang.env.COMPOSABLE_KERNEL_NOT_FOUND_MESSAGE = 'Composable Kernel is not installed or found in the expected path'¶: , which may lead to compilation bugs when utilize tilelang backend.

tilelang.env.TL_TEMPLATE_NOT_FOUND_MESSAGE = 'TileLang is not installed or found in the expected path'¶: , which may lead to compilation bugs when utilize tilelang backend.

tilelang.env.TVM_LIBRARY_NOT_FOUND_MESSAGE = 'TVM is not installed or found in the expected path'¶

class tilelang.env.CacheState¶

Class to manage global kernel caching state.

classmethod enable()¶: Enable kernel caching globally.

classmethod disable()¶: Disable kernel caching globally.

classmethod is_enabled()¶

Return current cache state.

Return type:: bool

class tilelang.env.EnvVar¶

Descriptor for managing access to a single environment variable.

Purpose¶

In many projects, access to environment variables is scattered across the codebase:

os.environ.get(…) calls are repeated everywhere
Default values are hard-coded in multiple places
Overriding env vars for tests/debugging is messy
There’s no central place to see all environment variables a package uses

This descriptor solves those issues by:

Centralizing the definition of the variable’s key and default value
Allowing dynamic reads from os.environ so changes take effect immediately
Supporting forced overrides at runtime (for unit tests or debugging)
Logging a warning when a forced value is used (helps detect unexpected overrides)
Optionally syncing forced values back to os.environ if global consistency is desired

How it works¶

This is a dataclass implementing the descriptor protocol (__get__, __set__)
When used as a class attribute, instance.attr triggers __get__()
→ returns either the forced override or the live value from os.environ
Assigning to the attribute (instance.attr = value) triggers __set__()
→ stores _forced_value for future reads
You may uncomment the os.environ[…] = value line in __set__ if you want the override to persist globally in the process

Example

```python class Environment:

TILELANG_PRINT_ON_COMPILATION = EnvVar(“TILELANG_PRINT_ON_COMPILATION”, “0”)

env = Environment() print(cfg.TILELANG_PRINT_ON_COMPILATION) # Reads from os.environ (with default fallback) cfg.TILELANG_PRINT_ON_COMPILATION = “1” # Forces value to “1” until changed/reset ```

Benefits¶

Centralizes all env-var keys and defaults in one place
Live, up-to-date reads (no stale values after import)
Testing convenience (override without touching the real env)
Improves IDE discoverability and type hints
Avoids hardcoding os.environ.get(…) in multiple places

key: str¶

default: str¶

get()¶

__get__(instance, owner)¶: Called when the attribute is accessed. 1. If a forced value is set, return it and log a warning 2. Otherwise, look up the value in os.environ; return the default if missing

__set__(instance, value)¶: Called when the attribute is assigned to. Stores the value as a runtime override (forced value). Optionally, you can also sync this into os.environ for global effect.

tilelang.env.enable_cache¶

tilelang.env.disable_cache¶

tilelang.env.is_cache_enabled¶

class tilelang.env.Environment¶

Environment configuration for TileLang. Handles CUDA/ROCm detection, integration paths, template/cache locations, auto-tuning configs, and build options.

CUDA_HOME¶

ROCM_HOME¶

TILELANG_PACKAGE_PATH¶

CUTLASS_INCLUDE_DIR¶

COMPOSABLE_KERNEL_INCLUDE_DIR¶

TVM_PYTHON_PATH¶

TVM_LIBRARY_PATH¶

TILELANG_TEMPLATE_PATH¶

TILELANG_CACHE_DIR¶

TILELANG_TMP_DIR¶

TILELANG_PRINT_ON_COMPILATION¶

TILELANG_CLEAR_CACHE¶

TILELANG_AUTO_TUNING_CPU_UTILITIES¶

TILELANG_AUTO_TUNING_CPU_COUNTS¶

TILELANG_AUTO_TUNING_MAX_CPU_COUNT¶

SKIP_LOADING_TILELANG_SO¶

TVM_IMPORT_PYTHON_PATH¶

is_cache_enabled()¶

Return type:: bool

enable_cache()¶

Return type:: None

disable_cache()¶

Return type:: None

is_print_on_compilation_enabled()¶

Return type:: bool

tilelang.env.env¶

tilelang.env.CUDA_HOME¶

tilelang.env.ROCM_HOME¶

tilelang.env.install_tvm_path¶

tilelang.env.install_cutlass_path¶

tilelang.env.install_ck_path¶

tilelang.env.install_tl_template_path¶

tilelang.env.CUTLASS_INCLUDE_DIR¶

tilelang.env.COMPOSABLE_KERNEL_INCLUDE_DIR¶

tilelang.env.TILELANG_TEMPLATE_PATH¶