gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ATIDLAS's python wrapper is now stand-alone

Browse Source
This commit is contained in:
Philippe Tillet
2014-10-15 04:24:25 -04:00
parent f91d3b422a
commit 30089b7fa6
13 changed files with 119 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/autotune/tools.py
View File

@@ -7,7 +7,6 @@ import sys
import pyopencl as cl
import pyviennacl as vcl
from pyviennacl.atidlas import StatementsTuple
class PhysicalLimitsNV:
def __init__(self, dev):
@@ -177,7 +176,7 @@ class OccupancyRecord:
def skip(template, statement, device):
statements = StatementsTuple(statement)
statements = vcl.pycore.StatementsTuple(statement)
registers_usage = template.registers_usage(statements)/4
lmem_usage = template.lmem_usage(statements)
local_size = template.parameters.local_size_0*template.parameters.local_size_1
@@ -187,7 +186,7 @@ def skip(template, statement, device):
return False
def benchmark(template, statement, device):
statements = StatementsTuple(statement)
statements = vcl.pycore.StatementsTuple(statement)
registers_usage = template.registers_usage(statements)/4
lmem_usage = template.lmem_usage(statements)
local_size = template.parameters.local_size_0*template.parameters.local_size_1