Merge branch 'd_dim_momenta_and_states' of github.com:Amps-GPU/BG-Tre…

…es into d_dim_momenta_and_states
Amps-GPU · Feb 7, 2025 · 46fd17c · 46fd17c
2 parents 8575db8 + 1be4e82
commit 46fd17c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bgtrees/finite_gpufields/operations.py b/bgtrees/finite_gpufields/operations.py
@@ -157,7 +157,7 @@ def ff_dot_product_tris(x, y, rank_x=None, rank_y=None):
     return ret.reshape_ff(shape_back)
 
 
-@tf.function(reduce_retracing=False, jit_compile=True)
+@tf.function(reduce_retracing=False)
 def ff_dot_product_tris_single_batch(x, y, rank_x=None, rank_y=None):
     """Single batched version of ff_dot_product_tris
     TODO: it should eventually go as well