Home
last modified time | relevance | path

Searched refs:buffer_addr (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/rl/
Dbuffer_append_cpu_kernel.h76 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
80 … if (memcpy_s(buffer_addr + IntToSize(index) * one_exp_len, one_exp_len * remain_size, exp_addr, in Launch()
85 … if (memcpy_s(buffer_addr, one_exp_len * remap_size, exp_addr, dist_len * remap_size) != EOK) { in Launch()
Dbuffer_get_cpu_kernel.h71 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
75 …if (memcpy_s(item_addr, one_exp_len, buffer_addr + IntToSize(index) * one_exp_len, dist_len) != EO… in Launch()
Dbuffer_sample_cpu_kernel.h96 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
100 …if (memcpy_s(output_addr + j * one_exp_len, one_exp_len, buffer_addr + index * one_exp_len, dist_l… in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/
Dbuffer_append_gpu_kernel.cc77 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
80 …BufferAppend(capacity_, one_exp_len, index_addr, LongToInt(exp_batch_), buffer_addr, exp_addr, cud… in Launch()
Dbuffer_get_gpu_kernel.cc78 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
81 BufferGetItem(one_exp_len, index_addr, one_exp_len, buffer_addr, item_addr, cuda_stream); in Launch()
Dbuffer_sample_gpu_kernel.cc116 auto buffer_addr = GetDeviceAddress<unsigned char>(inputs, i); in Launch() local
119 BufferSample(size, exp_element_list[i], indexes, buffer_addr, out_addr, cuda_stream); in Launch()
/third_party/mesa3d/src/compiler/nir/
Dnir_lower_printf.c51 nir_ssa_def *buffer_addr = nir_load_printf_buffer_address(b, ptr_bit_size); in lower_printf_instr() local
53 nir_build_deref_cast(b, buffer_addr, nir_var_mem_global, in lower_printf_instr()
/third_party/mesa3d/src/intel/tools/
Daubinator_viewer_decoder.cpp371 uint64_t buffer_addr = 0; in handle_3dstate_vertex_buffers() local
381 buffer_addr = vbs_iter.raw_value; in handle_3dstate_vertex_buffers()
382 vb = ctx_get_bo(ctx, true, buffer_addr); in handle_3dstate_vertex_buffers()
401 "buffer contents unavailable addr=0x%012" PRIx64, buffer_addr); in handle_3dstate_vertex_buffers()
423 uint64_t buffer_addr = 0; in handle_3dstate_index_buffer() local
433 buffer_addr = iter.raw_value; in handle_3dstate_index_buffer()
434 ib = ctx_get_bo(ctx, true, buffer_addr); in handle_3dstate_index_buffer()
443 buffer_addr); in handle_3dstate_index_buffer()
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_llvm_tess.c660 LLVMValueRef buffer_addr = get_tcs_tes_buffer_address( in si_copy_tcs_inputs() local
665 ac_build_buffer_store_dword(&ctx->ac, buffer, value, 4, buffer_addr, buffer_offset, 0, in si_copy_tcs_inputs()
/third_party/mesa3d/src/intel/vulkan/
Danv_nir_apply_pipeline_layout.c845 nir_ssa_def *buffer_addr = in lower_direct_buffer_instr() local
849 nir_ssa_def *bti = nir_channel(b, buffer_addr, 0); in lower_direct_buffer_instr()