add xlib.avecl

xlib/avecl/_internal/SCacheton.py

@@ -0,0 +1,55 @@
+class SCacheton:
+    """
+    Static class for caching classes and vars by hashable arguments
+    """
+    cachetons = {}
+    cachevars = {}
+
+    @staticmethod
+    def get(cls, *args, **kwargs):
+        """
+        Get class cached by args/kwargs
+        If it does not exist, creates new with *args,**kwargs
+        All cached data will be freed with cleanup()
+
+        You must not to store Tensor in SCacheton, use per-device cache vars
+        """
+        cls_multitons = SCacheton.cachetons.get(cls, None)
+        if cls_multitons is None:
+            cls_multitons = SCacheton.cachetons[cls] = {}
+
+        key = (args, tuple(kwargs.items()) )
+
+        data = cls_multitons.get(key, None)
+        if data is None:
+            data = cls_multitons[key] = cls(*args, **kwargs)
+
+        return data
+
+    @staticmethod
+    def set_var(var_name, value):
+        """
+        Set data cached by var_name
+        All cached data will be freed with cleanup()
+
+        You must not to store Tensor in SCacheton, use per-device cache vars
+        """
+        SCacheton.cachevars[var_name] = value
+
+    @staticmethod
+    def get_var(var_name):
+        """
+        Get data cached by var_name
+        All cached data will be freed with cleanup()
+
+        You must not to store Tensor in SCacheton, use per-device cache vars
+        """
+        return SCacheton.cachevars.get(var_name, None)
+
+    @staticmethod
+    def cleanup():
+        """
+        Free all cached objects
+        """
+        SCacheton.cachetons = {}
+        SCacheton.cachevars = {}