Searched refs:dynamic_kernel (Results 1 – 4 of 4) sorted by relevance
760 device::DynamicKernelPtr dynamic_kernel = nullptr; in LaunchKernelDynamic() local765 dynamic_kernel = gpu_kernel->DynamicKernel(); in LaunchKernelDynamic()768 if (dynamic_kernel && dynamic_kernel->is_dynamic_shape()) { in LaunchKernelDynamic()769 dynamic_kernel->InferShape(); in LaunchKernelDynamic()770 dynamic_kernel->UpdateArgs(); in LaunchKernelDynamic()795 … if (gpu_kernel != nullptr && dynamic_kernel != nullptr && dynamic_kernel->is_dynamic_shape()) { in LaunchKernelDynamic()880 device::DynamicKernelPtr dynamic_kernel = nullptr; in RunOpLaunchKernelDynamic() local885 dynamic_kernel = gpu_kernel->DynamicKernel(); in RunOpLaunchKernelDynamic()888 if (dynamic_kernel && dynamic_kernel->is_dynamic_shape()) { in RunOpLaunchKernelDynamic()889 dynamic_kernel->InferShape(); in RunOpLaunchKernelDynamic()[all …]
438 auto dynamic_kernel = kernel_mod->GenDynamicKernel(cnode, stream_); in GenDynamicKernel() local439 if (dynamic_kernel == nullptr) { in GenDynamicKernel()442 dynamic_kernel->Initialize(); in GenDynamicKernel()443 dynamic_kernels.emplace_back(dynamic_kernel); in GenDynamicKernel()893 for (const auto &dynamic_kernel : dynamic_kernels) { in RunDynamicKernelAsync() local894 MS_EXCEPTION_IF_NULL(dynamic_kernel); in RunDynamicKernelAsync()895 …if (dynamic_kernel->have_depends() || dynamic_kernel->GetKernelType() == KernelType::HCCL_KERNEL) { in RunDynamicKernelAsync()903 if (dynamic_kernel->is_dynamic_shape()) { in RunDynamicKernelAsync()904 dynamic_kernel->InferShape(); in RunDynamicKernelAsync()905 dynamic_kernel->UpdateArgs(); in RunDynamicKernelAsync()[all …]
384 device::DynamicKernelPtr dynamic_kernel = gpu_kernel->DynamicKernel(); in UpdateDynamicShape() local385 MS_EXCEPTION_IF_NULL(dynamic_kernel); in UpdateDynamicShape()387 dynamic_kernel->InferShape(); in UpdateDynamicShape()388 dynamic_kernel->UpdateArgs(); in UpdateDynamicShape()
2 … "kernel_info.cc" "executor/dynamic_kernel.cc" "executor/executor_callback.cc" "kernel_runtime.cc"