From 2b7ee5fb5a20fadfa24766442d2d3609ed291b63 Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Thu, 11 Aug 2011 17:41:48 -0400 Subject: Tell CUDA to enlarge L1 cache. --- gpu.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'gpu.py') diff --git a/gpu.py b/gpu.py index 16e6f4a..fbf1fef 100644 --- a/gpu.py +++ b/gpu.py @@ -72,6 +72,9 @@ class GPU(object): device = cuda.Device(device_id) self.context = device.make_context() print 'device %s' % self.context.get_device().name() + + self.context.set_cache_config(cuda.func_cache.PREFER_L1) + cuda_options = ['-I' + src.dir, '--use_fast_math', '--ptxas-options=-v'] self.module = SourceModule(src.kernel, options=cuda_options, no_extern_c=True) -- cgit