From 6eaa80be7a42ac0dbb770533463d9da1b1e11f3d Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Fri, 10 Sep 2010 12:53:40 -0400 Subject: [PATCH] Added property ctx.warps_per_cta --- cuburnlib/cuda.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cuburnlib/cuda.py b/cuburnlib/cuda.py index 09329cd..71222c1 100644 --- a/cuburnlib/cuda.py +++ b/cuburnlib/cuda.py @@ -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)