mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-11-03 18:00:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added property ctx.warps_per_cta

Browse Source
This commit is contained in:
Steven Robertson
2010-09-10 12:53:40 -04:00
parent 2f3ac42153
commit 6eaa80be7a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cuburnlib/cuda.py
View File

@ -50,6 +50,10 @@ class LaunchContext(object):
def threads_per_cta(self): def threads_per_cta(self):
return self.block[0] * self.block[1] * self.block[2] return self.block[0] * self.block[1] * self.block[2]
@property
def warps_per_cta(self):
return self.threads_per_cta / 32
def compile(self, verbose=False, **kwargs): def compile(self, verbose=False, **kwargs):
kwargs['ctx'] = self kwargs['ctx'] = self
self.ptx = PTXModule(self.entry_types, kwargs, self.build_tests) self.ptx = PTXModule(self.entry_types, kwargs, self.build_tests)