mirror of
https://github.com/stevenrobertson/cuburn.git
synced 2025-02-05 11:40:04 -05:00
Added property ctx.warps_per_cta
This commit is contained in:
parent
2f3ac42153
commit
6eaa80be7a
@ -50,6 +50,10 @@ class LaunchContext(object):
|
||||
def threads_per_cta(self):
|
||||
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):
|
||||
kwargs['ctx'] = self
|
||||
self.ptx = PTXModule(self.entry_types, kwargs, self.build_tests)
|
||||
|
Loading…
Reference in New Issue
Block a user