def test_create_program(self):
brig_file = get_brig_file()
symbol = '&__vector_copy_kernel'
brig_module = BrigModule.from_file(brig_file)
program = Program()
program.add_module(brig_module)
code = program.finalize(self.gpu.isa)
ex = Executable()
ex.load(self.gpu, code)
ex.freeze()
sym = ex.get_symbol(self.gpu, symbol)
self.assertGreater(sym.kernarg_segment_size, 0)
def test_create_program(self):
brig_file = get_brig_file()
symbol = '&__vector_copy_kernel'
brig_module = BrigModule.from_file(brig_file)
program = Program()
program.add_module(brig_module)
code = program.finalize(self.gpu.isa)
ex = Executable()
ex.load(self.gpu, code)
ex.freeze()
sym = ex.get_symbol(self.gpu, symbol)
self.assertGreater(sym.kernarg_segment_size, 0)
def test_loading_from_file(self):
arytype = types.float32[:]
kernel = compiler.compile_kernel(copy_kernel_1d, [arytype] * 2)
# Write the brig file out
brig_file = tempfile.NamedTemporaryFile(delete=False)
with brig_file:
brig_file.write(kernel.binary)
# Load BRIG file
symbol = '&{0}'.format(kernel.entry_name)
brig_module = BrigModule.from_file(brig_file.name)
# Cleanup
os.unlink(brig_file.name)
self._check(brig_module, symbol)
def test_from_file(self):
brig_file = get_brig_file()
brig_module = BrigModule.from_file(brig_file)
self.assertGreater(len(brig_module), 0)
def test_from_file(self):
brig_file = get_brig_file()
brig_module = BrigModule.from_file(brig_file)
self.assertGreater(len(brig_module), 0)
