diff options
author | Stan Seibert <stan@mtrr.org> | 2011-09-13 16:10:04 -0400 |
---|---|---|
committer | Stan Seibert <stan@mtrr.org> | 2011-09-13 16:10:04 -0400 |
commit | 4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75 (patch) | |
tree | 90a910177777115b6ef7ad2780b09ca69ff87109 /gpu.py | |
parent | 1ed35a8420b73afa76c12c82f4aebc31132650d9 (diff) | |
parent | 6f0703602270d03f4025221f13fee21aa842a863 (diff) | |
download | chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.gz chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.bz2 chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.zip |
merge
Diffstat (limited to 'gpu.py')
-rw-r--r-- | gpu.py | 25 |
1 files changed, 0 insertions, 25 deletions
@@ -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() |