Unverified Commit c411def2 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[torch.compile] fix shape specialization (#10722)


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent 308cc5e2
...@@ -2151,7 +2151,7 @@ class CompilationConfig(BaseModel): ...@@ -2151,7 +2151,7 @@ class CompilationConfig(BaseModel):
use_inductor: bool = True use_inductor: bool = True
inductor_specialize_for_cudagraph_no_more_than: Optional[int] = None inductor_specialize_for_cudagraph_no_more_than: Optional[int] = None
inductor_compile_sizes: Optional[List[int]] = Field(default_factory=dict) inductor_compile_sizes: Optional[List[int]] = Field(default=None)
inductor_compile_config: Dict = Field(default_factory=dict) inductor_compile_config: Dict = Field(default_factory=dict)
inductor_passes: Dict[str, str] = Field(default_factory=dict) inductor_passes: Dict[str, str] = Field(default_factory=dict)
...@@ -2290,9 +2290,8 @@ class CompilationConfig(BaseModel): ...@@ -2290,9 +2290,8 @@ class CompilationConfig(BaseModel):
if x <= self.inductor_specialize_for_cudagraph_no_more_than if x <= self.inductor_specialize_for_cudagraph_no_more_than
] ]
else: else:
assert self.inductor_compile_sizes is not None, ( if self.inductor_compile_sizes is None:
"inductor_compile_sizes should not be None when " self.inductor_compile_sizes = []
"inductor_specialize_for_cudagraph_no_more_than is None")
self.compile_sizes = self.inductor_compile_sizes self.compile_sizes = self.inductor_compile_sizes
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment