Merge + spotless

Author: ldematte

x-pack/plugin/gpu/src/main/java/org/elasticsearch/xpack/gpu/codec/CuVSResourceManager.java

@@ -8,9 +8,7 @@
 package org.elasticsearch.xpack.gpu.codec;
 
 import com.nvidia.cuvs.CuVSResources;
-
 import com.nvidia.cuvs.GPUInfoProvider;
-
 import com.nvidia.cuvs.spi.CuVSProvider;
 
 import org.elasticsearch.core.Strings;
@@ -83,8 +81,8 @@ class PoolingCuVSResourceManager implements CuVSResourceManager {
         static final GPUInfoProvider gpuInfoProvider = CuVSProvider.provider().gpuInfoProvider();
         static final PoolingCuVSResourceManager INSTANCE = new PoolingCuVSResourceManager(
             MAX_RESOURCES,
-            res ->gpuInfoProvider.getCurrentInfo(res).totalDeviceMemoryInBytes(),
-            res ->gpuInfoProvider.getCurrentInfo(res).freeDeviceMemoryInBytes(),
+            res -> gpuInfoProvider.getCurrentInfo(res).totalDeviceMemoryInBytes(),
+            res -> gpuInfoProvider.getCurrentInfo(res).freeDeviceMemoryInBytes(),
             PoolingCuVSResourceManager::getGpuUtilizationPercent
         );
 
@@ -104,7 +102,7 @@ public PoolingCuVSResourceManager(
             GpuInfoFunction totalMemoryInBytesProvider,
             GpuInfoFunction freeMemoryInBytesProvider,
             GpuInfoFunction gpuUtilizationPercentProvider
-            ) {
+        ) {
             this.totalMemoryInBytesProvider = totalMemoryInBytesProvider;
             this.freeMemoryInBytesProvider = freeMemoryInBytesProvider;
             this.gpuUtilizationPercentProvider = gpuUtilizationPercentProvider;
@@ -193,7 +191,7 @@ public ManagedCuVSResources acquire(int numVectors, int dims) throws Interrupted
         }
 
         private long estimateRequiredMemory(int numVectors, int dims) {
-            return (long)(GPU_COMPUTATION_MEMORY_FACTOR * numVectors * dims * Float.BYTES);
+            return (long) (GPU_COMPUTATION_MEMORY_FACTOR * numVectors * dims * Float.BYTES);
         }
 
         private static int getGpuUtilizationPercent(CuVSResources resources) {

x-pack/plugin/gpu/src/test/java/org/elasticsearch/xpack/gpu/codec/CuVSResourceManagerTests.java

@@ -8,7 +8,6 @@
 package org.elasticsearch.xpack.gpu.codec;
 
 import com.nvidia.cuvs.CuVSResources;
-
 import com.nvidia.cuvs.CuVSResourcesInfo;
 import com.nvidia.cuvs.GPUInfo;
 import com.nvidia.cuvs.GPUInfoProvider;
@@ -145,11 +144,7 @@ static class MockPoolingCuVSResourceManager extends CuVSResourceManager.PoolingC
         }
 
         private MockPoolingCuVSResourceManager(int capacity, List<Long> allocationList) {
-            super(
-                capacity,
-                res -> TOTAL_DEVICE_MEMORY_IN_BYTES,
-                res ->freeMemoryFunction(allocationList),
-                res -> 50);
+            super(capacity, res -> TOTAL_DEVICE_MEMORY_IN_BYTES, res -> freeMemoryFunction(allocationList), res -> 50);
             this.allocations = allocationList;
         }
 
@@ -165,8 +160,8 @@ protected CuVSResources createNew() {
         @Override
         public ManagedCuVSResources acquire(int numVectors, int dims) throws InterruptedException {
             var res = super.acquire(numVectors, dims);
-            long memory = (long)(numVectors * dims * Float.BYTES *
-                CuVSResourceManager.PoolingCuVSResourceManager.GPU_COMPUTATION_MEMORY_FACTOR);
+            long memory = (long) (numVectors * dims * Float.BYTES
+                * CuVSResourceManager.PoolingCuVSResourceManager.GPU_COMPUTATION_MEMORY_FACTOR);
             allocations.add(memory);
             log.info("Added [{}]", memory);
             return res;