From 513953c931a6a860eab314981ac486c762d443f3 Mon Sep 17 00:00:00 2001 From: Jean Kossaifi Date: Tue, 15 Nov 2022 09:41:23 -0800 Subject: [PATCH] Revert "Adds missing caching" This reverts commit 8000f982b875f70684fb1a40e92ca901684fe1a1. --- tensorly/tenalg/einsum_tenalg/caching.py | 20 -------------------- 1 file changed, 20 deletions(-) delete mode 100644 tensorly/tenalg/einsum_tenalg/caching.py diff --git a/tensorly/tenalg/einsum_tenalg/caching.py b/tensorly/tenalg/einsum_tenalg/caching.py deleted file mode 100644 index 6fb48bf75..000000000 --- a/tensorly/tenalg/einsum_tenalg/caching.py +++ /dev/null @@ -1,20 +0,0 @@ -EINSUM_PATH_CACHE = dict() - - -def einsum_path_cached(fun): - def wrapped(key, *args, **kwargs): - name = fun.__name__ - try: - cache = EINSUM_PATH_CACHE[name] - except KeyError: - EINSUM_PATH_CACHE[name] = dict() - cache = EINSUM_PATH_CACHE[name] - try: - equation = cache[key] - except KeyError: - equation = fun(*args, **kwargs) - cache[key] = equation - - return equation - - return wrapped