Merge branch 'dion/shapeopt-demo' of github.com:pasteurlabs/tesseract-jax into dion/shapeopt-demo

dionhaefner · dionhaefner · commit 13d115b545a2 · 2025-06-18T23:14:09.000+02:00
diff --git a/tesseract_jax/primitive.py b/tesseract_jax/primitive.py
@@ -229,7 +229,7 @@ def _dispatch(*args: ArrayLike) -> Any:
         array_args,
         ctx.avals_in,
         ctx.avals_out,
-        has_side_effect=False,
+        has_side_effect=True,
     )
     ctx.module_context.add_keepalive(keepalive)
     return result

Original file line number	Diff line number	Diff line change
`@@ -229,7 +229,7 @@ def _dispatch(*args: ArrayLike) -> Any:`
`229`	`229`	`array_args,`
`230`	`230`	`ctx.avals_in,`
`231`	`231`	`ctx.avals_out,`
`232`		`- has_side_effect=False,`
	`232`	`+ has_side_effect=True,`
`233`	`233`	`)`
`234`	`234`	`ctx.module_context.add_keepalive(keepalive)`
`235`	`235`	`return result`