We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d71b5c6 commit ac339d0Copy full SHA for ac339d0
helion/runtime/settings.py
@@ -53,7 +53,10 @@ class _Settings:
53
default_factory=list
54
)
55
index_dtype: torch.dtype = torch.int32
56
- dot_precision: Literal["tf32", "tf32x3", "ieee"] = "tf32"
+ dot_precision: Literal["tf32", "tf32x3", "ieee"] = cast(
57
+ "Literal['tf32', 'tf32x3', 'ieee']",
58
+ os.environ.get("TRITON_F32_DEFAULT", "tf32"),
59
+ )
60
static_shapes: bool = False
61
use_default_config: bool = os.environ.get("HELION_USE_DEFAULT_CONFIG", "0") == "1"
62
autotune_log_level: int = logging.INFO
0 commit comments