StableHLO 狀態

從 MHLO 啟動 StableHLO 時,我們沿用了 MHLO 的許多項目實作,包括美化、驗證和形狀推論。多虧了這個做法,我們針對這個障礙已大量涵蓋,但您還有很多方法可以檢查現有的實作項目是否完整,以及提供沒有不存在的實作。

這份即時文件適用於開發人員和使用者,可追蹤各方面各方面 (規格、驗證、類型推論、美化排版、口譯等) 的進度。

使用方式

如對應資料列中所述,系統會使用下列其中一個追蹤標籤,追蹤 StableHLO 運算的進度 (如對應資料列所述)。

  • 一般標籤
    • yes:是一種全方位的導入做法。
    • no:沒有實作,但會努力達成發展藍圖。請注意,ODS 已導入某些驗證作業,因此驗證器不得標示為「否」。
  • 驗證器和類型推論的自訂標籤
    • :有實作,且與 StableHLO 語意保持同步。
    • yes*:有實作,且與 XLA 語意保持同步。由於 XLA 語意通常處於未記載的狀態,因此我們使用 hlo_verifier.ccshape_inference.cc 做為參考。
    • revisit:有實作但不會顯示「yes」或「yes*」,原因是我們尚未稽核,或是我們發現有問題。
    • 不可行:沒有實作,因為這樣不合適。舉例來說,由於運算的結果類型無法透過其運算元和屬性推論出。

狀態

StableHLO Op 規格 驗證 型別推斷 美化印刷 翻譯模式
abs
add
after_all
all_gather 回顧
all_reduce 回顧
all_to_all 回顧
以及
atan2
batch_norm_grad 回顧 回顧
batch_norm_inference 回顧 回顧
batch_norm_training 回顧 回顧
bitcast_convert 不可行
廣播訊息 是* 是* 回顧
broadcast_in_dim 不可行
客服案件 回顧
Cbrt
水泥
白洞 回顧
限制取值範圍 回顧
collective_permute 回顧
compare
複雜
compute_reshape_shape 回顧
concatenate
常數
完成轉換 不可行
卷積 不可行 回顧
餘弦
count_leading_zeros
create_token 是* 是* 回顧
跨備用資源總和 回顧 是* 回顧
cstr_reshapable 回顧
custom_call 不可行
點號 回顧 不可行 回顧
dot_general 回顧 不可行
dynamic_broadcast_in_dim 回顧 不可行
dynamic_conv 回顧
dynamic_gather 回顧 回顧
dynamic_iota 回顧 不可行
dynamic_pad 回顧
dynamic_reshape 回顧 不可行
dynamic_slice
dynamic_update_slice
einsum 回顧 回顧
指數
exponential_minus_one
fft 回顧
floor
收集
get_dimension_size
get_tuple_element
if 回顧
想像力
動態內廣告 不可行
Iota 不可行
is_finite
紀錄/記錄檔
log_plus_one
物流
地圖 回顧
最高
最低
negate
不是
optimization_barrier
外動態
墊片
partition_id
罌粟紅
功率
real
real_dynamic_slice 回顧
Recv 不可行
減少 回顧 回顧
reduce_precision
reduce_scatter 回顧
reduce_window 回顧
剩餘
replica_id
重塑 不可行
return 回顧 不可行
反向排序
奈米 回顧
rng_bit_generator 回顧 不可行 回顧
round_nearest_afz
round_nearest_even
rsqrt
scatter 回顧
選取
select_and_scatter 回顧
傳送
set_dimension_size 是* 是*
shift_left
shift_right_arithmetic
shift_right_logical
簽署
正弦
配量
排序
sqrt
subtract
坦赫
torch_index_select 回顧 回顧
trace 回顧 回顧
轉置
triangular_solve 回顧 回顧
tuple
unary_einsum 回顧 回顧
uniform_dequantize
uniform_quantize 回顧 不可行
回顧 回顧
xor