From d3f90fb8a6cb1da4f192fc058daee3699055fcf3 Mon Sep 17 00:00:00 2001 From: blepping Date: Fri, 12 Apr 2024 12:39:14 -0600 Subject: [PATCH] BlehBlockOps: Fix skip targeting. --- py/nodes/ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/nodes/ops.py b/py/nodes/ops.py index f2eaf99..73a75b0 100644 --- a/py/nodes/ops.py +++ b/py/nodes/ops.py @@ -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