Trace global memory usage in histogramm tests

Author: antonwolfy

dpnp/tests/third_party/cupy/statistics_tests/test_histogram.py

@@ -1,6 +1,7 @@
 import sys
 import unittest
 
+import dpctl
 import numpy
 import pytest
 
@@ -44,6 +45,20 @@ def for_all_dtypes_combination_bincount(names):
     return testing.for_dtypes_combination(_all_types, names=names)
 
 
+# TODO: comments
+_dev = dpctl.select_default_device()
+
+
+@pytest.fixture(autouse=True)
+def setup_each():
+    print("\n[Setup] Run before each test")
+    free_mem = dpctl.utils.intel_device_info(_dev).get("free_memory", None)
+    if free_mem:
+        print(f"Global memory available: {free_mem}")
+    yield
+    print("[Teardown] Run after each test")
+
+
 class TestHistogram(unittest.TestCase):
 
     @testing.for_all_dtypes(no_bool=True, no_complex=True)