Searched refs:block_dim_ (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.h | 39 block_dim_(block_dim), in AiCoreDynamicKernel()
47 block_dim_(block_dim), in AiCoreDynamicKernel()
66 uint32_t block_dim_; variable
|
| D | ai_core_dynamic_kernel.cc | 56 …if (RT_ERROR_NONE != rtKernelLaunchWithHandle(handle_, dev_func.c_str(), block_dim_, runtime_args_… in Execute()
61 …if (RT_ERROR_NONE != rtKernelLaunch(stub_func_, block_dim_, runtime_args_.data(), args_size, l2ctr… in Execute()
143 block_dim_ = op_run_info_v2.GetBlockDim(); in ComputeTiling()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | ascend_kernel_mod.h | 35 uint32_t block_dim() { return block_dim_; } in block_dim()
48 uint32_t block_dim_{1};
|
| D | common_utils.h | 51 uint32_t block_dim_; member
|
| /third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/ |
| D | gpu_data_saver.h | 33 std::string block_dim_; member
53 …event.basic_info_->op_name << ',' << event.basic_info_->stream_id << ',' << event.block_dim_ << ','
|
| D | gpu_data_saver.cc | 33 block_dim_ = basic_info_->activity_type == ActivityType::kKernel in ActivityData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 98 auto funcstub = KernelManager::GenFuncStub(*kernel_pack_, false, &block_dim_); in GenTask()
105 MS_LOG(INFO) << "block_dim is:" << block_dim_; in GenTask()
108 …unique_name_, stream_id, stub_func, block_dim_, args, 0, sm_desc, nullptr, 0, meta_data, input_dat… in GenTask()
|
| D | tbe_utils.h | 73 uint32_t block_dim_; member
|
| D | tbe_utils.cc | 300 *block_dim = kernelmeta->block_dim_; in GenFuncStub()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/ |
| D | profiling_desc.h | 40 … : ProfDesc(std::move(op_name)), task_id_(task_id), block_dim_(block_dim), stream_id_(stream_id) {} in TaskDesc()
46 uint32_t block_dim_; variable
|
| D | profiling_desc.cc | 27 .append(std::to_string(block_dim_)) in ToString()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 74 block_dim_(block_dim), in TbeTaskInfo()
87 uint32_t block_dim() const { return block_dim_; } in block_dim()
105 uint32_t block_dim_; variable
|