-interpreter-instrument-with-probe

Menyisipkan operasi interpreter.probe dalam program StableHLO.

Menelusuri program StableHLO dan memasukkan instrumentasi pemeriksaan operasi setelah setiap operasi yang sesuai (lihat di bawah untuk mengetahui bagaimana ditentukan). Instrumentasi digunakan untuk mengekstrak nilai tensor dari penafsir referensi StableHLO untuk perbandingan nanti dengan runtime lainnya.

Semua operasi dianggap cocok untuk instrumentasi, kecuali konstanta ops yang tidak memiliki nilai tensor dengan nilai apa pun (yaitu menghasilkan tuple atau token atau tidak ada nilai yang ditampilkan, tidak akan diinstrumentasikan). Operasi yang sesuai akan diinstrumentasikan terlepas dari tingkat secara bertingkat. Artinya, operasi di dalam region loop/cabang juga akan diinstrumentasi.

Operasi berinstrumen akan memiliki nilai yang ditampilkan yang ditulis ke {i>disk<i} menggunakan format data NumPy saat dieksekusi. Jika useDebugInfo lulus diaktifkan, informasi debug lokasi akan digunakan bila tersedia untuk secara unik mengidentifikasi nilai tensor berinstrumen (yaitu pass akan mengekstrak probe_id dari NamedLoc(probe_id@<...>) dan gunakan format probe_id.#). Jika tidak, nilai berinstrumen akan dirujuk dalam kenaikan urutan: probe1, probe2, ... Lihat interpreter.probe untuk mengetahui informasi tentang bagaimana data diserialisasi.

Opsi

-useDebugInfo : Whether or not to use location debug data as `probe_id` values.