Create Symbol with is_prim=True in _register_custom_op (#2516)

crcrpar · web-flow · commit bf1fcf2f66c0 · 2025-09-16T11:18:28.000+02:00
diff --git a/thunder/torch/custom_op.py b/thunder/torch/custom_op.py
@@ -357,7 +357,7 @@ def _register_custom_op(custom_op: CustomOpDef) -> Symbol:
         name=fn_name,
         meta=meta_fn,
         id=op_id,
-        is_prim=False,
+        is_prim=True,
         tags=tags,
     )
     # Register both `torch.ops.my_lib.foo` and `torch.ops.my_lib.foo.default`.

Original file line number	Diff line number	Diff line change
`@@ -357,7 +357,7 @@ def _register_custom_op(custom_op: CustomOpDef) -> Symbol:`
`357`	`357`	`name=fn_name,`
`358`	`358`	`meta=meta_fn,`
`359`	`359`	`id=op_id,`
`360`		`- is_prim=False,`
	`360`	`+ is_prim=True,`
`361`	`361`	`tags=tags,`
`362`	`362`	`)`
`363`	`363`	# Register both `torch.ops.my_lib.foo` and `torch.ops.my_lib.foo.default`.