BlehBlockOps: Fix skip targeting.

2024-04-12 12:39:14 -06:00
parent e350413601
commit d3f90fb8a6
1 changed files with 1 additions and 1 deletions
@@ -283,7 +283,7 @@ class Operation:
                    amount = int(t.shape[dims[0]] * amount)
                out = torch.roll(t, amount, dims=dims)
            case OpType.TARGET_SKIP:
-                if get(state, "hsp") is None:
+                if state.get("hsp") is None:
                    if state["target"] == "hsp":
                        state["target"] = "h"
                    return