xla
a2a4ff4c - Bugfix + add more test for llama (#5439)

Commit
2 years ago
Bugfix + add more test for llama (#5439) Bugfix details: 1. When the graph have mutations the exported graph will have additional inputs. For now we are dropping them. 2. We should trace with args instead of final_args.
Author
Parents
Loading