Add a supported check for cuVS - can be used in more tests eventually

Author: ChrisHegarty

qa/full-cluster-restart/src/javaRestTest/java/org/elasticsearch/upgrades/FullClusterRestartIT.java

@@ -1004,7 +1004,11 @@ public void testSnapshotRestore() throws IOException {
             if (minimumIndexVersion().before(IndexVersions.V_8_0_0) && randomBoolean()) {
                 settings.put(IndexSettings.INDEX_SOFT_DELETES_SETTING.getKey(), randomBoolean());
             }
-            createIndex(index, settings.build());
+            System.out.println("HEGO settings=" + settings);
+            var resp = createIndex(index, settings.build());
+            // Map<String, Object> snapResponse = entityAsMap(resp);
+            System.out.println("HEGO createIndex response=" + resp);
+
             indexRandomDocuments(count, true, true, randomBoolean(), i -> jsonBuilder().startObject().field("field", "value").endObject());
         } else {
             count = countOfIndexedRandomDocuments();
@@ -1094,9 +1098,18 @@ public void testSnapshotRestore() throws IOException {
             (builder, params) -> builder.field("indices", index)
         );
         createSnapshot.addParameter("wait_for_completion", "true");
-        client().performRequest(createSnapshot);
+        System.out.println("HEGO createSnapshot=" + createSnapshot);
+
+        var resp = client().performRequest(createSnapshot);
+
+        System.out.println("HEGO createSnapshot resp=" + resp);
+
+        System.out.println("HEGO Build.current().version()=" + Build.current().version());
+        System.out.println("HEGO IndexVersion.current()=" + IndexVersion.current());
+        System.out.println("HEGO getOldClusterVersion=" + getOldClusterVersion());
+        System.out.println("HEGO getOldClusterIndexVersion=" + getOldClusterIndexVersion());
 
-        checkSnapshot("old_snap", count, getOldClusterVersion(), getOldClusterIndexVersion());
+        checkSnapshot("old_snap", count, getOldClusterVersion(), getOldClusterIndexVersion());  // HERE
         if (false == isRunningAgainstOldCluster()) {
             checkSnapshot("new_snap", count, Build.current().version(), IndexVersion.current());
         }
@@ -1282,6 +1295,7 @@ private void checkSnapshot(String snapshotName, int count, String tookOnVersion,
         // Check the snapshot metadata, especially the version
         Request listSnapshotRequest = new Request("GET", "/_snapshot/repo/" + snapshotName);
         Map<String, Object> snapResponse = entityAsMap(client().performRequest(listSnapshotRequest));
+        System.out.println("HEGO listSnapshotRequest response=" + snapResponse);
 
         assertEquals(singletonList(snapshotName), XContentMapValues.extractValue("snapshots.snapshot", snapResponse));
         assertEquals(singletonList("SUCCESS"), XContentMapValues.extractValue("snapshots.state", snapResponse));
@@ -1290,6 +1304,7 @@ private void checkSnapshot(String snapshotName, int count, String tookOnVersion,
         // which could affect the top range of the index release version
         String firstReleaseVersion = tookOnIndexVersion.toReleaseVersion().split("-")[0];
         assertThat(
+            // HERE
             (Iterable<String>) XContentMapValues.extractValue("snapshots.version", snapResponse),
             anyOf(
                 contains(tookOnVersion),

x-pack/plugin/gpu/src/main/java/module-info.java

@@ -7,6 +7,7 @@
 
 /** Provides GPU-accelerated support for vector search. */
 module org.elasticsearch.gpu {
+    requires org.elasticsearch.logging;
     requires org.apache.lucene.core;
     requires org.elasticsearch.xcontent;
     requires org.elasticsearch.server;

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

@@ -7,6 +7,8 @@
 
 package org.elasticsearch.xpack.gpu.codec;
 
+import com.nvidia.cuvs.CuVSResources;
+
 import org.apache.lucene.codecs.KnnVectorsFormat;
 import org.apache.lucene.codecs.KnnVectorsReader;
 import org.apache.lucene.codecs.KnnVectorsWriter;
@@ -16,6 +18,8 @@
 import org.apache.lucene.codecs.perfield.PerFieldKnnVectorsFormat;
 import org.apache.lucene.index.SegmentReadState;
 import org.apache.lucene.index.SegmentWriteState;
+import org.elasticsearch.logging.LogManager;
+import org.elasticsearch.logging.Logger;
 
 import java.io.IOException;
 
@@ -25,6 +29,8 @@
  */
 public class GPUVectorsFormat extends KnnVectorsFormat {
 
+    private static final Logger LOG = LogManager.getLogger(GPUVectorsFormat.class);
+
     public static final String NAME = "GPUVectorsFormat";
     public static final String GPU_IDX_EXTENSION = "gpuidx";
     public static final String GPU_META_EXTENSION = "mgpu";
@@ -69,4 +75,20 @@ static GPUVectorsReader getGPUReader(KnnVectorsReader vectorsReader, String fiel
         }
         return null;
     }
+
+    /** Tells whether the platform supports cuvs. */
+    public static boolean supported() {
+        try (var resources = CuVSResources.create()) {
+            return true;
+        } catch (UnsupportedOperationException uoe) {
+            var msg = uoe.getMessage() == null ? "" : ": " + uoe.getMessage();
+            LOG.warn("cuvs is not supported on this platform or java version" + msg);
+        } catch (Throwable t) {
+            if (t instanceof ExceptionInInitializerError ex) {
+                t = ex.getCause();
+            }
+            LOG.warn("Exception occurred during creation of cuvs resources. " + t);
+        }
+        return false;
+    }
 }

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

@@ -14,6 +14,7 @@
 import org.apache.lucene.tests.index.BaseKnnVectorsFormatTestCase;
 import org.apache.lucene.tests.util.TestUtil;
 import org.elasticsearch.common.logging.LogConfigurator;
+import org.junit.BeforeClass;
 
 public class GPUVectorsFormatTests extends BaseKnnVectorsFormatTestCase {
 
@@ -22,6 +23,11 @@ public class GPUVectorsFormatTests extends BaseKnnVectorsFormatTestCase {
         LogConfigurator.configureESLogging(); // native access requires logging to be initialized
     }
 
+    @BeforeClass
+    public static void beforeClass() {
+        assumeTrue("cuvs not supported", GPUVectorsFormat.supported());
+    }
+
     static final Codec codec = TestUtil.alwaysKnnVectorsFormat(new GPUVectorsFormat());
 
     @Override