Searched refs:kernel_args (Results 1 – 5 of 5) sorted by relevance
94 auto kernel_args = absl::make_unique<se::KernelArgsArray<kKernelArgsLimit>>(); in ExecuteOnStream() local97 kernel_args->add_device_memory_argument(buf); in ExecuteOnStream()105 *kernel_args)) { in ExecuteOnStream()
1646 std::unordered_map<const BufferAllocation*, llvm::Value*> kernel_args; in BuildKernelThunk() local1651 kernel_args[*buffers_it] = arg_it; in BuildKernelThunk()1673 loc = InBoundsGEP(kernel_args.at(slice.allocation()), in BuildKernelThunk()1692 bindings_.SetTempBufferBase(kernel_args.at(*temp_buffer)); in BuildKernelThunk()
618 uint32_t *kernel_args; in si_upload_compute_input() local634 kernel_args = (uint32_t*)kernel_args_ptr; in si_upload_compute_input()639 kernel_args[i] = info->grid[i]; in si_upload_compute_input()640 kernel_args[i + 3] = info->grid[i] * info->block[i]; in si_upload_compute_input()641 kernel_args[i + 6] = info->block[i]; in si_upload_compute_input()645 memcpy(kernel_args + (num_work_size_bytes / 4), info->input, in si_upload_compute_input()651 kernel_args[i]); in si_upload_compute_input()
15 setenv kernel_args "setenv bootargs ${console} root=/dev/mmcblk${mmcrootdev}p${mmcrootpart} rootfst…72 run kernel_args;
878 KernelArgsArray<sizeof...(args)> kernel_args; in ThenLaunch() local879 kernel.PackParams(&kernel_args, args...); in ThenLaunch()882 parent_->Launch(this, thread_dims, block_dims, kernel, kernel_args); in ThenLaunch()