Skip to content

Commit 6b3f61a

Browse files
authored
Bypass type check on two functionalized graph method calls.
1 parent 8089054 commit 6b3f61a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/shark_turbine/aot/builtins/jittable.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,12 +213,12 @@ def flat_wrapped_f(*args):
213213
if "functorch_functionalize" in self._passes:
214214
transformed_f = functorch_functionalize(transformed_f, *flat_pytorch_args)
215215

216-
for node in transformed_f.graph.nodes:
216+
for node in transformed_f.graph.nodes: # type: ignore
217217
if node.op == "call_function":
218218
if node.target == torch._ops.ops.aten.lift_fresh_copy.default:
219219
print(f"replaced lift_fresh_copy")
220220
node.target = torch._ops.ops.aten.clone.default
221-
transformed_f.recompile()
221+
transformed_f.recompile() # type: ignore
222222

223223
# Ask dynamo to give us an aten graph.
224224
# TODO: Cache this for repeated calls.

0 commit comments

Comments
 (0)