mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-01 00:15:30 +08:00
22bb84fa9d
issue: https://github.com/milvus-io/milvus/issues/29230 this pr do these things: 1. add gpu brute force; 2. limit gpu index only support l2 / ip; Signed-off-by: cqy123456 <qianya.cheng@zilliz.com>
78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
// 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.
|
|
|
|
package indexparamcheck
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/cockroachdb/errors"
|
|
)
|
|
|
|
type IndexCheckerMgr interface {
|
|
GetChecker(indexType string) (IndexChecker, error)
|
|
}
|
|
|
|
// indexCheckerMgrImpl implements IndexChecker.
|
|
type indexCheckerMgrImpl struct {
|
|
checkers map[IndexType]IndexChecker
|
|
once sync.Once
|
|
}
|
|
|
|
func (mgr *indexCheckerMgrImpl) GetChecker(indexType string) (IndexChecker, error) {
|
|
mgr.once.Do(mgr.registerIndexChecker)
|
|
|
|
adapter, ok := mgr.checkers[indexType]
|
|
if ok {
|
|
return adapter, nil
|
|
}
|
|
return nil, errors.New("Can not find conf adapter: " + indexType)
|
|
}
|
|
|
|
func (mgr *indexCheckerMgrImpl) registerIndexChecker() {
|
|
mgr.checkers[IndexRaftIvfFlat] = newRaftIVFFlatChecker()
|
|
mgr.checkers[IndexRaftIvfPQ] = newRaftIVFPQChecker()
|
|
mgr.checkers[IndexRaftCagra] = newCagraChecker()
|
|
mgr.checkers[IndexRaftBruteForce] = newRaftBruteForceChecker()
|
|
mgr.checkers[IndexFaissIDMap] = newFlatChecker()
|
|
mgr.checkers[IndexFaissIvfFlat] = newIVFBaseChecker()
|
|
mgr.checkers[IndexFaissIvfPQ] = newIVFPQChecker()
|
|
mgr.checkers[IndexScaNN] = newScaNNChecker()
|
|
mgr.checkers[IndexFaissIvfSQ8] = newIVFSQChecker()
|
|
mgr.checkers[IndexFaissBinIDMap] = newBinFlatChecker()
|
|
mgr.checkers[IndexFaissBinIvfFlat] = newBinIVFFlatChecker()
|
|
mgr.checkers[IndexHNSW] = newHnswChecker()
|
|
mgr.checkers[IndexDISKANN] = newDiskannChecker()
|
|
}
|
|
|
|
func newIndexCheckerMgr() *indexCheckerMgrImpl {
|
|
return &indexCheckerMgrImpl{
|
|
checkers: make(map[IndexType]IndexChecker),
|
|
}
|
|
}
|
|
|
|
var indexCheckerMgr IndexCheckerMgr
|
|
|
|
var getIndexCheckerMgrOnce sync.Once
|
|
|
|
// GetIndexCheckerMgrInstance gets the instance of IndexCheckerMgr.
|
|
func GetIndexCheckerMgrInstance() IndexCheckerMgr {
|
|
getIndexCheckerMgrOnce.Do(func() {
|
|
indexCheckerMgr = newIndexCheckerMgr()
|
|
})
|
|
return indexCheckerMgr
|
|
}
|