enhance: Set the default GPU version autoindex to use the CAgra index (#41906)

issue:  #41907

Signed-off-by: yusheng.ma <[email protected]>

Author: presburger

internal/util/indexparamcheck/constraints.go

@@ -39,8 +39,8 @@ const (
 	CagraGraphDegree = "graph_degree"
 	CagraBuildAlgo   = "build_algo"
 
-	CargaBuildAlgoIVFPQ     = "IVF_PQ"
-	CargaBuildAlgoNNDESCENT = "NN_DESCENT"
+	CagraBuildAlgoIVFPQ     = "IVF_PQ"
+	CagraBuildAlgoNNDESCENT = "NN_DESCENT"
 
 	// Sparse Index Param
 	SparseDropRatioBuild = "drop_ratio_build"
@@ -64,7 +64,7 @@ var (
 	BinIvfMetrics             = []string{metric.HAMMING, metric.JACCARD}                                             // const
 	HnswMetrics               = []string{metric.L2, metric.IP, metric.COSINE}                                        // const
 	RaftMetrics               = []string{metric.L2, metric.IP}
-	CagraBuildAlgoTypes       = []string{CargaBuildAlgoIVFPQ, CargaBuildAlgoNNDESCENT}
+	CagraBuildAlgoTypes       = []string{CagraBuildAlgoIVFPQ, CagraBuildAlgoNNDESCENT}
 	supportDimPerSubQuantizer = []int{32, 28, 24, 20, 16, 12, 10, 8, 6, 4, 3, 2, 1}              // const
 	supportSubQuantizer       = []int{96, 64, 56, 48, 40, 32, 28, 24, 20, 16, 12, 8, 4, 3, 2, 1} // const
 	SparseMetrics             = []string{metric.IP, metric.BM25}                                 // const

pkg/util/paramtable/autoindex_param.go

@@ -86,13 +86,7 @@ func (p *AutoIndexConfig) init(base *BaseTable) {
 	}
 	p.EnableResultLimitCheck.Init(base.mgr)
 
-	p.IndexParams = ParamItem{
-		Key:          "autoIndex.params.build",
-		Version:      "2.2.0",
-		DefaultValue: `{"M": 18,"efConstruction": 240,"index_type": "HNSW", "metric_type": "COSINE"}`,
-		Formatter:    GetBuildParamFormatter(FloatVectorDefaultMetricType, "autoIndex.params.build"),
-		Export:       true,
-	}
+	p.IndexParams = GetIndexParam()
 	p.IndexParams.Init(base.mgr)
 
 	p.SparseIndexParams = ParamItem{

pkg/util/paramtable/autoindex_param_cuda.go

@@ -0,0 +1,30 @@
+// Licensed to the LF AI & Data foundation under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//go:build cuda
+// +build cuda
+
+package paramtable
+
+func GetIndexParam() ParamItem {
+	return ParamItem{
+		Key:          "autoIndex.params.build",
+		Version:      "2.2.0",
+		DefaultValue: `{"intermediate_graph_degree":64, "graph_degree": 32, "index_type": "GPU_CAGRA", "metric_type": "COSINE"}`,
+		Formatter:    GetBuildParamFormatter(FloatVectorDefaultMetricType, "autoIndex.params.build"),
+		Export:       true,
+	}
+}

pkg/util/paramtable/autoindex_param_nocuda.go

@@ -0,0 +1,30 @@
+// Licensed to the LF AI & Data foundation under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//go:build !cuda
+// +build !cuda
+
+package paramtable
+
+func GetIndexParam() ParamItem {
+	return ParamItem{
+		Key:          "autoIndex.params.build",
+		Version:      "2.2.0",
+		DefaultValue: `{"M": 18,"efConstruction": 240,"index_type": "HNSW", "metric_type": "COSINE"}`,
+		Formatter:    GetBuildParamFormatter(FloatVectorDefaultMetricType, "autoIndex.params.build"),
+		Export:       true,
+	}
+}