diff options
| author | Stan Seibert <stan@mtrr.org> | 2011-08-11 17:41:48 -0400 |
|---|---|---|
| committer | Stan Seibert <stan@mtrr.org> | 2011-08-11 17:41:48 -0400 |
| commit | 2b7ee5fb5a20fadfa24766442d2d3609ed291b63 (patch) | |
| tree | 1c98fc3667b7e1e8ee9b31efd7b778ef8b0a0617 | |
| parent | 1b9aef380e629c136e045412621483ebb4a4164f (diff) | |
| download | chroma-2b7ee5fb5a20fadfa24766442d2d3609ed291b63.tar.gz chroma-2b7ee5fb5a20fadfa24766442d2d3609ed291b63.tar.bz2 chroma-2b7ee5fb5a20fadfa24766442d2d3609ed291b63.zip | |
Tell CUDA to enlarge L1 cache.
| -rw-r--r-- | gpu.py | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -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) |