summaryrefslogtreecommitdiff
path: root/gpu.py
diff options
context:
space:
mode:
authorStan Seibert <stan@mtrr.org>2011-09-13 16:10:04 -0400
committerStan Seibert <stan@mtrr.org>2011-09-13 16:10:04 -0400
commit4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75 (patch)
tree90a910177777115b6ef7ad2780b09ca69ff87109 /gpu.py
parent1ed35a8420b73afa76c12c82f4aebc31132650d9 (diff)
parent6f0703602270d03f4025221f13fee21aa842a863 (diff)
downloadchroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.gz
chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.bz2
chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.zip
merge
Diffstat (limited to 'gpu.py')
-rw-r--r--gpu.py25
1 files changed, 0 insertions, 25 deletions
diff --git a/gpu.py b/gpu.py
index b98d42b..4e2ecf3 100644
--- a/gpu.py
+++ b/gpu.py
@@ -797,28 +797,3 @@ def create_cuda_context(device_id=None):
context.set_cache_config(cuda.func_cache.PREFER_L1)
return context
-
-class GPU(object):
- def __init__(self, device_id=None):
- """Initialize a GPU context on the specified device.
- If device_id is None, the default device is used."""
-
- if device_id is None:
- self.context = pycuda.tools.make_default_context()
- else:
- device = cuda.Device(device_id)
- self.context = device.make_context()
-
- print 'device %s' % self.context.get_device().name()
-
- self.print_mem_info()
-
- self.context.set_cache_config(cuda.func_cache.PREFER_L1)
-
- def print_mem_info(self):
- free, total = cuda.mem_get_info()
-
- print '%.1f %% of device memory is free.' % ((free/float(total))*100)
-
- def __del__(self):
- self.context.pop()