Commits


Abhishek Jindal authored and GitHub committed 2f93d97fd02
Add cuda visible devices for Mistral benchmark (#18764) ### Description <!-- Describe your changes. --> Add cuda visible devices for Mistral benchmark as it is not working for Torch compile and throwing an error. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> Error: File "/opt/conda/envs/ptca/lib/python3.8/site-packages/torch/_inductor/triton_heuristics.py", line 556, in run return launcher( File "<string>", line 8, in launcher RuntimeError: Triton Error [CUDA]: invalid device context