chore: cleanup garbage code (#29770)
This commit is contained in:
@@ -131,7 +131,7 @@ def run_baseline(inp):
|
||||
|
||||
def run_packed(inp):
|
||||
"""Packed path: single fused kernel directly on mixed_qkv."""
|
||||
B, HV, K, V = inp["B"], inp["HV"], inp["K"], inp["V"]
|
||||
B, HV, V = inp["B"], inp["HV"], inp["V"]
|
||||
ssm_states = inp["ssm_states"].clone()
|
||||
out = inp["mixed_qkv"].new_empty(B, 1, HV, V)
|
||||
|
||||
@@ -173,7 +173,7 @@ def check_correctness(B, H, HV, K, V, pool_size, device, dtype, seed=42):
|
||||
try:
|
||||
torch.testing.assert_close(o_packed, o_baseline, atol=atol, rtol=rtol)
|
||||
output_ok = True
|
||||
except AssertionError as e:
|
||||
except AssertionError:
|
||||
output_ok = False
|
||||
out_diff = (o_packed - o_baseline).abs().max().item()
|
||||
|
||||
|
||||
@@ -171,7 +171,7 @@ if __name__ == "__main__":
|
||||
"--limit",
|
||||
type=int,
|
||||
default=None,
|
||||
help="If set, only evaluate this many samples (debug / smoke runs).",
|
||||
help="If set, only evaluate this many samples (debug / quick runs).",
|
||||
)
|
||||
EvalArgs.add_cli_args(parser)
|
||||
args = parser.parse_args()
|
||||
|
||||
Reference in New Issue
Block a user