Make mypy opt-out instead of opt-in (#33205)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -570,7 +570,7 @@ class CompilationConfig:
|
||||
pass_config: PassConfig = field(default_factory=PassConfig)
|
||||
"""Custom inductor passes, see PassConfig for more details"""
|
||||
|
||||
max_cudagraph_capture_size: int | None = field(default=None)
|
||||
max_cudagraph_capture_size: int = field(default=None)
|
||||
"""The maximum cudagraph capture size.
|
||||
|
||||
If cudagraph_capture_sizes is specified, this will be set to the largest
|
||||
@@ -743,6 +743,7 @@ class CompilationConfig:
|
||||
"level",
|
||||
"mode",
|
||||
"cudagraph_mode",
|
||||
"max_cudagraph_capture_size",
|
||||
"use_inductor_graph_partition",
|
||||
mode="wrap",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user