Checklist Penerjemah StableHLO

Dalam dokumen ini, kami merangkum panduan untuk mengimplementasikan dan meninjau op untuk penafsir. Kami sengaja menyertakan beberapa item tindakan tambahan terkait pemverifikasi dan inferensi jenis, dengan tujuan untuk membuat progres di antarmuka tersebut bersama implementasi penafsir.

Saat menerapkan pengoperasian

  1. Sediakan strategi pengujian yang ditulis secara eksplisit (dalam deskripsi PR) yang serupa dengan ini untuk digunakan sebagai referensi saat meninjau metode inferensi verifikasi dan jenis, serta pengujian yang sesuai. Peninjau akan memeriksa ulang apakah deskripsi sudah komprehensif.
  2. Lihat hlo_evaluator untuk mengidentifikasi detail implementasi yang rumit dan potensi celah fungsionalitas.
  3. Ajukan tiket untuk komponen software yang sesuai jika Anda menemukan bug atau fungsi yang hilang.

Setelah menerapkan pengoperasian

  1. Di StablehloOps.td:

    1. Pastikan summary di ODS pengoperasian mengikuti format standar. (tiket terkait)
    2. Tambahkan komentar yang merujuk ke label batasan (misalnya Cn atau In) dari spesifikasi dalam format xyz_cn atau xyz_in, untuk operasi XyzOp, guna mengidentifikasi korespondensi antara batasan dalam ODS dan spesifikasi. Contoh berikut menunjukkan cara menambahkan label batasan sebagai komentar bersama mlir Traits dan TypeConstraints. Perhatikan bahwa xyz_c4 mengacu pada batasan yang ditentukan dalam class StableHLO_FooOp (misalnya StableHLO_ShapedInterfaceOp, StableHLO_UnaryElementwiseOp, StableHLO_Op, dll.).

       def StableHLO_XyzOp: StableHLO_FooOp<"xyz", [Trait1,
           Trait2 /*xyz_c1, xyz_c2*/, InferTensorType /*xyz_c3*/]> { /*xyz_c4*/
            ...
         let summary = "Xyz operation";
         let arguments = (ins
            1DTensorOf<[HLO_Float]>:$a, /*xyz_c5, xyz_i1*/
            HLO_Tensor:$b, /*xyz_i2*/
            ....
         );
      );
      
  2. Di TypeInference.cpp dan StablehloOps.cpp:

    1. Hapus komentar yang menyatakan hal-hal seperti "Verifikasi properti berikut: ...".
    2. Tambahkan komentar yang merujuk pada label batasan (misalnya Cn atau In) dari spesifikasi dalam format xyz_cn atau xyz_in, untuk operasi XyzOp, guna mengidentifikasi bagian pemverifikasi dan fungsi bentuk mana yang sesuai dengan batasan dalam spesifikasi.
      1. Anda dapat membuat komentar dengan beberapa label batasan atau memiliki beberapa komentar dengan label batasan yang sama. Ini semua tergantung pada bagaimana batasan tersebut diimplementasikan. Jika ada batasan berturut-turut, ringkas sebagai xyz_cn...xyz_cm, xyz_in...xyz_jn.
      2. Jika ada ketidakcocokan antara batasan dalam implementasi VS dan batasan dalam spesifikasi, pastikan ada masalah terbuka yang mencerminkan perbedaan tersebut.
  3. Dalam pengujian penerjemah:

    1. Tambahkan file bernama interpret_<op_mnemonic>.mlir.
    2. Tulis pengujian dengan mengikuti pedoman pengujian.
  4. Dalam direktori testdata:

    1. Menjalankan pengujian nonaktif yang dicakup oleh operasi yang baru ditambahkan.
    2. Jika pengujian lulus, aktifkan pengujian dengan mengonversi RUN-DISABLED menjadi RUN.
    3. Jika pengujian gagal karena alasan selain ketidakcocokan presisi, perbaiki implementasi/pengujian.
    4. Untuk ketidakcocokan presisi, beri tag pada pengujian dengan RUN-DISABLED(#1278) (jika belum dilakukan).
  5. Di ops_stablehlo.mlir:

    1. Pastikan setidaknya ada satu pengujian (positif atau negatif) untuk setiap batasan dalam metode inferensi jenis dan pemverifikasi; batasan yang dicakup dalam ODS tidak akan diuji. Pengujian ini sebagian besar akan bernilai negatif, yang menguji bahwa batasan tidak terpenuhi atau positif, menguji bahwa bentuk yang disimpulkan sudah benar.
    2. Pastikan semua pengujian yang terkait dengan operasi yang sedang diuji ditempatkan bersama.
    3. Pastikan semua pengujian yang terkait dengan operasi yang sedang diuji diawali dengan makro CHECK-LABEL yang menyala.
    4. Pilih nama fungsi pengujian menggunakan format xyz_cn_im_... untuk batasan pengujian fungsi Cn, Im, dll. untuk op XyzOp. Jika format yang diusulkan tidak berlaku, pertahankan nama yang ada.
    5. Setelah langkah di atas selesai, urutkan semua pengujian yang terkait dengan operasi yang sedang diuji menurut abjad berdasarkan nama fungsi.
    6. Terus tambahkan pengujian hingga ccov menampilkan cakupan >= 90% untuk operasi tersebut.
  6. Pada infer_stablehlo.mlir:

    1. Pastikan semua batasan yang terkait dengan pengujian inferensi bentuk ada dalam file ini, dengan mengikuti panduan penamaan yang sama seperti yang disebutkan di atas.
    2. Pindahkan pengujian inferensi bentuk dari file ops_stablehlo.mlir ke dalam file ini.
  7. Di spec.md:

    1. Tambahkan link ke interpret_<op_mnemonic>.mlir ke bagian "Contoh" (misalnya, Contoh Lainnya).
    2. Pastikan spesifikasi hanya memiliki 1 contoh.
    3. Pastikan contoh spesifikasi mengikuti panduan pengujian.
    4. Pastikan pengujian contoh spesifikasi dapat ditafsirkan.
    5. Pastikan contoh spesifikasinya sama dengan yang ada di ODS.
  8. Di status.md:

    1. Perbarui kolom "Penerjemah" menjadi yes.