xla
Add stack frame id metadata and correct filename and line number for custom op_name
#5838
Merged

Add stack frame id metadata and correct filename and line number for custom op_name #5838

qihqi merged 14 commits into pytorch:master from master
mrnikwaws
mrnikwaws Add python binding to allow custom op_name metadata for lowere HLO
51a15d91
mrnikwaws Lint fix for clang-format
7d392522
mrnikwaws Lint fix for customer op_name test code
b9eb7dc6
mrnikwaws Requested nit changes
c5f74b12
mrnikwaws As discussed increase timeout on GPU tests by 20%
457dd85d
mrnikwaws Merge branch 'pytorch:master' into master
b951da48
mrnikwaws Add lowering for stack frame index and stack frame id in metadata
6b72c56d
JackCaoG
mrnikwaws Add fix for stack depth when using set custom op_name in a python con…
20532a91
mrnikwaws Changes after adding tests for lowered stack frames and finding sever…
b266fd4d
mrnikwaws Lint fixes
3a03d23b
mrnikwaws
qihqi qihqi requested a review from qihqi qihqi 1 year ago
mrnikwaws Cleanup old code / comments
a16b9cf1
mrnikwaws Add routine to XlaNode to search back through operands and recusively…
ef0dba1c
mrnikwaws
mrnikwaws Fix recursion condition so we don't explore nodes with metadata
a2d92bf4
qihqi
qihqi commented on 2023-11-28
qihqi qihqi requested a review from qihqi qihqi 1 year ago
qihqi
qihqi commented on 2023-11-29
mrnikwaws Refactor stack frame builder into a separate class as requested by re…
b17e700d
mrnikwaws
qihqi
qihqi approved these changes on 2023-12-01
qihqi qihqi merged 5fa89223 into master 1 year ago
alanwaketan
alanwaketan commented on 2023-12-07

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone