Test cupti fix (#54) Summary: ## What does this PR do? Fixes flaky test https://app.circleci.com/pipelines/github/facebookresearch/HolisticTraceAnalysis/144/workflows/2bf5d461-8f9f-4b15-9f7e-c7df57f61d62/jobs/678 ## Before submitting - [ ] Was this discussed/approved via a Github issue? (no need for typos, doc improvements) - [x] N/A - [ ] Did you write any new necessary tests? - [x] N/A - [ ] Did you make sure to update the docs? - [x] N/A - [ ] Did you update the [changelog](https://github.com/facebookresearch/HolisticTraceAnalysis/blob/main/CHANGELOG.md)? - [x] N/A Pull Request resolved: https://github.com/facebookresearch/HolisticTraceAnalysis/pull/54 Reviewed By: briancoutinho Differential Revision: D45995642 Pulled By: anupambhatnagar fbshipit-source-id: afe9decabf0d7c3c73fe089344cb01411045b2f6