refactor(attn): init hisparse_coordinator before attn_backend; replace lazy property with init-time capture (#26012)
Co-authored-by: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Sonnet 4.6
parent
c5251a98a9
commit
d765dfd043
@@ -91,6 +91,7 @@ class MockModelRunner:
|
||||
device=self.device,
|
||||
enable_memory_saver=False,
|
||||
)
|
||||
self.hisparse_coordinator = None
|
||||
|
||||
|
||||
@unittest.skipIf(not torch.cuda.is_available(), "Test requires CUDA")
|
||||
|
||||
@@ -76,6 +76,7 @@ class MockModelRunner:
|
||||
device=self.device,
|
||||
enable_memory_saver=False,
|
||||
)
|
||||
self.hisparse_coordinator = None
|
||||
|
||||
|
||||
class MockReqToTokenPool:
|
||||
|
||||
@@ -268,6 +268,7 @@ class MockModelRunner:
|
||||
device=self.device,
|
||||
enable_memory_saver=False,
|
||||
)
|
||||
self.hisparse_coordinator = None
|
||||
|
||||
|
||||
def compare_outputs(trtllm_out, reference_out, tolerance=1e-2):
|
||||
|
||||
@@ -252,6 +252,7 @@ class MockModelRunner:
|
||||
"dsa_decode_backend": "fa3",
|
||||
},
|
||||
)()
|
||||
self.hisparse_coordinator = None
|
||||
|
||||
|
||||
@unittest.skipIf(not torch.cuda.is_available(), "Test requires CUDA")
|
||||
|
||||
Reference in New Issue
Block a user