Skip to content

Commit ac339d0

Browse files
authored
Set default dot_precision to TRITON_F32_DEFAULT (#197)
Fixes #196
1 parent d71b5c6 commit ac339d0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

helion/runtime/settings.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,10 @@ class _Settings:
5353
default_factory=list
5454
)
5555
index_dtype: torch.dtype = torch.int32
56-
dot_precision: Literal["tf32", "tf32x3", "ieee"] = "tf32"
56+
dot_precision: Literal["tf32", "tf32x3", "ieee"] = cast(
57+
"Literal['tf32', 'tf32x3', 'ieee']",
58+
os.environ.get("TRITON_F32_DEFAULT", "tf32"),
59+
)
5760
static_shapes: bool = False
5861
use_default_config: bool = os.environ.get("HELION_USE_DEFAULT_CONFIG", "0") == "1"
5962
autotune_log_level: int = logging.INFO

0 commit comments

Comments
 (0)