Return test check

Author: yuslepukhin

onnxruntime/core/providers/cuda/cuda_mempool_arena.h

@@ -23,7 +23,7 @@ namespace cuda {
  *        a single process is hosting more than one cuda session. This arena hosts cuda memory pool
  *        which has some tunable parameters to control its memory usage and de-allocates memory back to
  *        the device according to the specified params. This is opposite to the BFCArena which only
- *        attempts to free memory on Shrink() at the end of the run.
+ *        attempts to free memory on Shrink() if configured at the end of the run.
  *
  * ### Behavior
  * - Creates a **process-local** CUDA mempool for a specific device (from `OrtMemoryInfo`).

onnxruntime/test/providers/cuda/cuda_mempool_arena_test.cc

@@ -52,6 +52,23 @@ static bool IsCudaMemPoolSupported() {
     return false;
   }
 
+  // Creating a cuda mempool in some pipelines fails with
+  // CUDA failure 801: operation not supported ; GPU=0 ; hostname=af14bbb1c000000 ;
+  // Even though CUDA version may be 12.8 possibly due to the driver.
+  cudaMemPoolProps props{};
+  // Pinned is not the same as pinned allocator, cudaMemLocationTypeDevice actually does not exist
+  // even though is present in some internet docs.
+  props.allocType = cudaMemAllocationTypePinned;
+  props.handleTypes = cudaMemHandleTypeNone;        // local to process
+  props.location.type = cudaMemLocationTypeDevice;  // Device memory
+  props.location.id = 0;                            // test device 0
+  cudaMemPool_t pool;
+  auto cuda_error = cudaMemPoolCreate(&pool, &props);
+  if (cuda_error != cudaSuccess) {
+    return false;
+  }
+  cuda_error = cudaMemPoolDestroy(pool);
+
   return true;
 }