From cdd834dc9431306ab61c39974e7efef822dfc5a7 Mon Sep 17 00:00:00 2001
From: asglover <140220574+asglover@users.noreply.github.com>
Date: Sat, 12 Sep 2026 00:51:18 -0700
Subject: [PATCH] lru cache for jinja env

---
 openequivariance/openequivariance/templates/jinja_utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/openequivariance/openequivariance/templates/jinja_utils.py b/openequivariance/openequivariance/templates/jinja_utils.py
index b4ede597..076fd198 100644
--- a/openequivariance/openequivariance/templates/jinja_utils.py
+++ b/openequivariance/openequivariance/templates/jinja_utils.py
@@ -1,3 +1,5 @@
+from functools import lru_cache
+
 from jinja2 import Environment, PackageLoader
 
 
@@ -16,6 +18,7 @@ def sizeof(dtype):
         raise Exception("Provided undefined datatype to sizeof!")
 
 
+@lru_cache(maxsize=2)
 def get_jinja_environment(is_hip=False):
     env = Environment(
         loader=PackageLoader("openequivariance"), extensions=["jinja2.ext.do"]