Clean up CUDA state between tests

Author: rraminen

test/test_cuda.py

@@ -442,6 +442,10 @@ def test_out_of_memory_retry(self):
         IS_JETSON, "oom reporting has issues on jetson igx due to partial nvml support"
     )
     def test_set_per_process_memory_fraction(self):
+        if torch.version.hip:
+           torch.cuda.empty_cache()
+           torch.cuda.reset_peak_memory_stats()
+
         orig = torch.cuda.get_per_process_memory_fraction(0)
         try:
             # test invalid fraction value.