xla
5fa89223 - Add stack frame id metadata and correct filename and line number for custom op_name (#5838)

Commit
1 year ago
Add stack frame id metadata and correct filename and line number for custom op_name (#5838) * Add python binding to allow custom op_name metadata for lowere HLO * As discussed increase timeout on GPU tests by 20% * Add lowering for stack frame index and stack frame id in metadata * Add fix for stack depth when using set custom op_name in a python context * Changes after adding tests for lowered stack frames and finding several issues * Add routine to XlaNode to search back through operands and recusively set meta data * Fix recursion condition so we don't explore nodes with metadata
Author
Parents
Loading