Method create_argument_buffer

compute/compute_kernel.cpp:55–93

}

unique_ptr<argument_buffer> compute_kernel::create_argument_buffer(const compute_queue& cqueue, const uint32_t& arg_index,
																   const COMPUTE_MEMORY_FLAG add_mem_flags) const {
	const auto& dev = cqueue.get_device();
	const auto entry = get_kernel_entry(dev);
	if (!entry || !entry->info) {
		log_error("no kernel entry/info for device $", dev.name);
		return {};
	}
	
	// need to take care of argument index translation when stage_input arguments exist
	uint32_t idx = 0;
	for (uint32_t actual_idx = 0, count = uint32_t(entry->info->args.size()); idx <= count; ++actual_idx) {
		if (idx >= count) {
			log_error("argument index is out-of-bounds: $", arg_index);
			return {};
		}
		
		const auto& arg_info = entry->info->args[idx];
		if (actual_idx == arg_index) {
			if (arg_info.special_type != llvm_toolchain::SPECIAL_TYPE::ARGUMENT_BUFFER) {
				log_error("argument #$ is not an argument buffer", arg_index);
				return {};
			}
			break;
		}
		
		if (arg_info.special_type == llvm_toolchain::SPECIAL_TYPE::STAGE_INPUT) {
			// skip to next actual arg
			while (idx < entry->info->args.size() &&
				   entry->info->args[idx].special_type == llvm_toolchain::SPECIAL_TYPE::STAGE_INPUT) {
				++idx;
			}
		} else {
			++idx;
		}
	}
	
	return create_argument_buffer_internal(cqueue, *entry, entry->info->args[idx], arg_index, idx, add_mem_flags);
}

unique_ptr<argument_buffer> compute_kernel::create_argument_buffer_internal(const compute_queue&,
																			const kernel_entry&,