Home
last modified time | relevance | path

Searched refs:kAttrAtomicAddMemSize (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_atomic_clean.cc112 AnfAlgo::SetNodeAttr(kAttrAtomicAddMemSize, MakeValue(clean_size), clean_node); in ConstructKernelGraphAndSetAttr()
Dkernel_build_ascend.cc180 AnfAlgo::SetNodeAttr(kAttrAtomicAddMemSize, MakeValue(clean_size), clear_zero); in AddTbeClearZeroNode()
206 AnfAlgo::SetNodeAttr(kAttrAtomicAddMemSize, MakeValue(clean_size_list), clear_zero); in AddFusionTbeClearZeroNode()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc137 auto clear_mems = AnfAlgo::GetNodeAttr<std::vector<size_t>>(anf_node_ptr, kAttrAtomicAddMemSize); in LaunchAddrCleanKernel()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h358 constexpr auto kAttrAtomicAddMemSize = "automic_add_mem_size"; variable