Searched refs:AscendLaunchAtomicClean (Results 1 – 3 of 3) sorted by relevance
24 void AscendLaunchAtomicClean::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem()26 size_t AscendLaunchAtomicClean::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel()30 uint8_t *AscendLaunchAtomicClean::AllocDeviceMem(size_t size) { return AscendLaunchKernel::AllocDev… in AllocDeviceMem()32 void AscendLaunchAtomicClean::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_grap… in KernelSelect()36 void AscendLaunchAtomicClean::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph… in KernelBuild()40 void AscendLaunchAtomicClean::LaunchOpKernel() { in LaunchOpKernel()72 void AscendLaunchAtomicClean::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem()77 std::shared_ptr<session::KernelGraph> AscendLaunchAtomicClean::ObtainAtomicCleanKernelGraph() { in ObtainAtomicCleanKernelGraph()94 void AscendLaunchAtomicClean::ConstructKernelGraphAndSetAttr() { in ConstructKernelGraphAndSetAttr()
25 class AscendLaunchAtomicClean : public AscendLaunchKernel {27 AscendLaunchAtomicClean(void *stream, TypeId dtype, size_t total_size) in AscendLaunchAtomicClean() function33 ~AscendLaunchAtomicClean() override = default;
183 std::make_shared<AscendLaunchAtomicClean>(compute_stream_, tensor_type_list_[0], total_size_); in CreateLaunchAtomicClean()