-interpreter-instrument-with-probe

แทรกการดำเนินการ interpreter.probe ในโปรแกรม StableHLO

เดินผ่านโปรแกรม StableHLO และแทรกการดำเนินการตรวจสอบหลังจากการดำเนินการที่เหมาะสมแต่ละรายการ (ดูวิธีกำหนดการดำเนินการที่เหมาะสมได้ที่ด้านล่าง) เครื่องมือวัดผลใช้เพื่อดึงค่าเทนเซอร์กลางๆ ออกจากโปรแกรมตีความข้อมูลอ้างอิง StableHLO เพื่อเปรียบเทียบกับรันไทม์อื่นๆ ในภายหลัง

การดำเนินการทั้งหมดถือว่าเหมาะสมสำหรับการใช้เครื่องมือ ยกเว้นการดำเนินการแบบคงที่ซึ่งไม่มีการส่งคืนค่า Tensor ใดๆ (เช่น การดำเนินการที่สร้าง Tuple หรือโทเค็น หรือไม่มีการวัดค่าส่งกลับ) ระบบจะเครื่องมือวัดการดำเนินการที่เหมาะสมโดยไม่คำนึงถึงระดับการฝัง กล่าวคือ ระบบจะตรวจสอบการดำเนินการภายในภูมิภาคของลูป/สาขาด้วย

ระบบจะเขียนค่าที่ส่งคืนของการดำเนินการที่มีเครื่องมือวัดไปยังดิสก์โดยใช้รูปแบบข้อมูล NumPy เมื่อดำเนินการ หากเปิดใช้ตัวเลือกuseDebugInfo pass ระบบจะใช้ข้อมูลการแก้ไขข้อบกพร่องของตำแหน่ง (หากมี) เพื่อระบุค่าเทนเซอร์ที่ตรวจสอบได้อย่างไม่ซ้ำกัน (กล่าวคือ pass จะดึงprobe_idจาก NamedLoc(probe_id@<...>) และใช้รูปแบบ probe_id.#) หากไม่ใช้ตัวเลือกนี้ ระบบจะอ้างอิงค่าที่ตรวจสอบตามลําดับที่เพิ่มขึ้น เช่น probe1, probe2, ... ดูข้อมูลเพิ่มเติมเกี่ยวกับวิธีจัดรูปแบบข้อมูลได้ที่ interpreter.probe

ตัวเลือก

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