Remove unnecessary assignment to num_tokens

It has been set above, and `effects` doesn't change. PiperOrigin-RevId: 538881671
harborn · Jun 8, 2023 · 9dccaad · 9dccaad
1 parent 1445149
commit 9dccaad
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/jax/_src/interpreters/mlir.py b/jax/_src/interpreters/mlir.py
@@ -846,7 +846,6 @@ def aval_to_types(aval):
     # If we aren't creating tokens they will be the initial inputs to the
     # MLIR function.
     output_token_types = []
-    num_tokens = len(effects)
     token_types = [token_type() for _ in effects]
   token_avals = [core.AbstractToken] * len(effects)
   input_avals = dim_var_avals + token_avals + jaxpr.in_avals