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)