MB-68591: GPU-Accelerated Vector Search

index_update.go

@@ -507,6 +507,9 @@ func compareFieldMapping(original, updated *mapping.FieldMapping) (*index.Update
 		if original.VectorIndexOptimizedFor != updated.VectorIndexOptimizedFor {
 			return nil, fmt.Errorf("vectorIndexOptimizedFor cannot be updated for vector and vector_base64 fields")
 		}
+		if original.UseGPU != updated.UseGPU {
+			return nil, fmt.Errorf("useGPU cannot be updated for vector and vector_base64 fields")
+		}
 	}
 	if original.IncludeInAll != updated.IncludeInAll {
 		return nil, fmt.Errorf("includeInAll cannot be changed")

mapping/field.go

@@ -83,6 +83,9 @@
 	VectorIndexOptimizedFor string `json:"vector_index_optimized_for,omitempty"`
 
 	SynonymSource string `json:"synonym_source,omitempty"`
+
+	// Flag that indicates whether to use GPU for field indexing and searching
+	UseGPU bool `json:"gpu,omitempty"`
 }
 
 // NewTextFieldMapping returns a default field mapping for text
@@ -226,6 +229,9 @@
 	if fm.SkipFreqNorm {
 		rv |= index.SkipFreqNorm
 	}
+	if fm.UseGPU {
+		rv |= index.GPU
+	}
 	return rv
 }
 
@@ -479,6 +485,11 @@
 			if err != nil {
 				return err
 			}
+		case "gpu":
+			err := util.UnmarshalJSON(v, &fm.UseGPU)
+			if err != nil {
+				return err
+			}
 		default:
 			invalidKeys = append(invalidKeys, k)
 		}

mapping/mapping_vectors.go

@@ -237,6 +237,11 @@ func validateVectorFieldAlias(field *FieldMapping, parentName string,
 				"(different similarity values %s and %s)", fieldAlias.Name,
 				field.Similarity, fieldAlias.Similarity)
 		}
+		if field.UseGPU != fieldAlias.UseGPU {
+			return fmt.Errorf("field: '%s', invalid alias "+
+				"(different useGPU values %v and %v)", fieldAlias.Name,
+				field.UseGPU, fieldAlias.UseGPU)
+		}
 
 		return nil
 	}