TBE實戰運算元開發 DSL方式

2021-10-08 23:06:02 字數 729 閱讀 3998

tbe運算元編譯過程

總結

def

sqrt_cce

(shape, dtype, kernel_name=

"sqrt_cce"

, need_build=

false

, need_print=

false

):

shape : tensor的屬性,表示tensor的形狀,用list或tuple型別表示。

例如(3, 2, 3)、(4, 10)。

- dtype : tensor的資料型別,用字串型別表示。

例如「float32」、「float16」、「int8」等。

以計算過程中的一條dsl語句舉例,

te.lang.cce.vexp(mul_val),含義為對tensor

(mul_val)求e^(x)

1.經過tbe框架的schedule過程,dsl描述轉變為

produce物件來表示計算過程,此時具有資料表示

含義;2.經過pass層,produce物件中數學算式替換為指令

描述;3.經過build cce 過程,produce 物件表示的計算過程

完全轉變為cce語句;

4.最後經過compile,cce**編譯輸出二進位制可執行文

件。

Ascend Pytorch運算元適配層開發

ascend pytorch運算元適配層開發 適配方法 找到和pytorch運算元功能對應的npu tbe運算元,根據運算元功能計算出輸出tensor的size,再根據tbe運算元原型構造對應的input output attr,傳遞給acl完成tbe運算元的執行。說明 tbe運算元實現的原始檔存放...

Django開發實戰

django開發流程 參考首先說明的是專案中要包含這幾個檔案 changelog.md 用來記錄專案的變更,主要針對每次發布版本的更新 如果團隊使用git,其實也可以使用git生成 另外,這個檔案對於開源專案,十分必要 如果有興趣的話可以看看django的release logreadme.md 用...

halcon學習筆記 (14)運算元 開發

dev clear obj objects 刪除乙個iconic物件 dev error var errorvar,mode 定義或取消定義乙個錯誤變數 dev get exception data exception,name value 獲取異常資料 exception 包含異常資料或使用者定義...