多func處理函數(shù)【免費(fèi)下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計(jì)算圖優(yōu)化、多流并行、內(nèi)存復(fù)用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對(duì) PyTorch、TensorFlow 前端的友好接入能力并同時(shí)支持 onnx、pb 等主流模型格式的解析與編譯。項(xiàng)目地址: https://gitcode.com/cann/ge產(chǎn)品支持情況產(chǎn)品是否支持Atlas A3 訓(xùn)練系列產(chǎn)品/Atlas A3 推理系列產(chǎn)品√Atlas A2 訓(xùn)練系列產(chǎn)品/Atlas A2 推理系列產(chǎn)品√函數(shù)功能用戶自定義的多flow func處理函數(shù)。函數(shù)原型普通場(chǎng)景下函數(shù)輸入由框架準(zhǔn)備完畢后直接給用戶使用用戶直接使用入?yún)⒅械膄lowMsg即可。std::functionint32_t (const std::shared_ptrMetaRunContext runContext, const std::vectorstd::shared_ptrFlowMsg flowMsg)流式輸入即函數(shù)入?yún)殛?duì)列場(chǎng)景下由用戶自行從輸入隊(duì)列中獲取數(shù)據(jù)使用。std::functionint32_t (const std::shared_ptrMetaRunContext runContext, const std::vectorstd::shared_ptrFlowMsgQueue flowMsgQueue)參數(shù)說(shuō)明參數(shù)名輸入/輸出描述runContext輸入處理函數(shù)的上下文信息。FlowMsg/FlowMsgQueue輸入函數(shù)的入?yún)?輸入隊(duì)列。返回值0SUCCESS。otherFAILED具體請(qǐng)參考UDF錯(cuò)誤碼。異常處理如果有不可恢復(fù)的異常信息發(fā)生返回ERROR。其他情況則調(diào)用SetRetcode設(shè)置輸出tensor的錯(cuò)誤碼。如果返回success調(diào)度會(huì)終止。約束說(shuō)明使用流式輸入flow func時(shí)需要在ProcessPoint編譯配置文件中將對(duì)應(yīng)func的stream_input字段設(shè)置為true。例如{ func_list: [ { func_name: Func, stream_input: true } ], input_num: 1, output_num: 1, target_bin: libfunc.so, workspace: ./, cmakelist_path: CMakeLists.txt, }【免費(fèi)下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計(jì)算圖優(yōu)化、多流并行、內(nèi)存復(fù)用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對(duì) PyTorch、TensorFlow 前端的友好接入能力并同時(shí)支持 onnx、pb 等主流模型格式的解析與編譯。項(xiàng)目地址: https://gitcode.com/cann/ge創(chuàng)作聲明：本文部分內(nèi)容由AI輔助生成（AIGC），僅供參考