Changeset 750ffa5 in sasmodels for sasmodels/kernelcl.py
- Timestamp:
- Mar 9, 2015 2:04:55 PM (9 years ago)
- Branches:
- master, core_shell_microgels, costrafo411, magnetic_model, release_v0.94, release_v0.95, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests
- Children:
- 3a45c2c
- Parents:
- 48f0194
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
sasmodels/kernelcl.py
r3c56da87 r750ffa5 68 68 #open(info['name']+'.cl','w').write(source) 69 69 #source = open(info['name']+'.cl','r').read() 70 return GpuModel(source, info, dtype)70 return GpuModel(source, info, np.dtype(dtype)) 71 71 72 72 ENV = None … … 92 92 Return the size of an execution batch for *kernel* running on *queue*. 93 93 """ 94 return kernel.get_work_group_info(cl.kernel_work_group_info.PREFERRED_WORK_GROUP_SIZE_MULTIPLE, 95 queue.device) 94 return kernel.get_work_group_info( 95 cl.kernel_work_group_info.PREFERRED_WORK_GROUP_SIZE_MULTIPLE, 96 queue.device) 96 97 97 98 def _stretch_input(vector, dtype, extra=1e-3, boundary=32): … … 219 220 self.info = info 220 221 self.source = source 221 self.dtype = dtype222 self.dtype = np.dtype(dtype) 222 223 self.program = None # delay program creation 223 224 … … 230 231 self.__dict__ = state.copy() 231 232 232 def __call__(self, input_value):233 if self.dtype != input_value.dtype:234 raise TypeError("data and kernel have different types")233 def __call__(self, q_input): 234 if self.dtype != q_input.dtype: 235 raise TypeError("data is %s kernel is %s" % (q_input.dtype, self.dtype)) 235 236 if self.program is None: 236 237 compiler = environment().compile_program 237 238 self.program = compiler(self.info['name'], self.source, self.dtype) 238 kernel_name = generate.kernel_name(self.info, input_value.is_2D)239 kernel_name = generate.kernel_name(self.info, q_input.is_2D) 239 240 kernel = getattr(self.program, kernel_name) 240 return GpuKernel(kernel, self.info, input_value)241 return GpuKernel(kernel, self.info, q_input) 241 242 242 243 def release(self):
Note: See TracChangeset
for help on using the changeset viewer.