From bd4a0d0cc59b80dca84471174b2928bbd4c1586e Mon Sep 17 00:00:00 2001
From: Stan Seibert <stan@mtrr.org>
Date: Tue, 31 Jan 2012 19:44:54 -0500
Subject: Import memoize decorator directly from pytools as it is not present
 in the old location in newer pycuda releases.

---
 chroma/gpu/tools.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/chroma/gpu/tools.py b/chroma/gpu/tools.py
index b151f80..34b4166 100644
--- a/chroma/gpu/tools.py
+++ b/chroma/gpu/tools.py
@@ -1,4 +1,5 @@
 import numpy as np
+import pytools
 import pycuda.tools
 from pycuda import characterize
 import pycuda.driver as cuda
@@ -32,7 +33,7 @@ def get_cu_module(name, options=None, include_source_directory=True):
     return pycuda.compiler.SourceModule(source, options=options,
                                         no_extern_c=True)
 
-@pycuda.tools.memoize
+@pytools.memoize
 def get_cu_source(name):
     """Get the source code for a CUDA source file located in the chroma cuda
     directory at src/[name]."""
-- 
cgit