@@ -89,7 +89,7 @@ class GPU_Quantizer {
89
89
90
90
// Make sure L2 is used, since other
91
91
if (metric != DistMetric::L2) {
92
- LOG (Helper::LogLevel::LL_Error, " Only L2 distance currently supported for PQ or OPQ\n " );
92
+ SPTAGLIB_LOG (Helper::LogLevel::LL_Error, " Only L2 distance currently supported for PQ or OPQ\n " );
93
93
exit (1 );
94
94
}
95
95
@@ -103,7 +103,7 @@ class GPU_Quantizer {
103
103
m_BlockSize = pq_quantizer->GetBlockSize ();
104
104
m_DimPerSubvector = pq_quantizer->GetDimPerSubvector ();
105
105
106
- LOG (Helper::LogLevel::LL_Debug, " Using PQ - numSubVectors:%d, KsPerSub:%ld, BlockSize:%ld, DimPerSub:%d, total size of tables:%ld\n " , m_NumSubvectors, m_KsPerSubvector, m_BlockSize, m_DimPerSubvector, m_BlockSize*m_NumSubvectors*sizeof (float ));
106
+ SPTAGLIB_LOG (Helper::LogLevel::LL_Debug, " Using PQ - numSubVectors:%d, KsPerSub:%ld, BlockSize:%ld, DimPerSub:%d, total size of tables:%ld\n " , m_NumSubvectors, m_KsPerSubvector, m_BlockSize, m_DimPerSubvector, m_BlockSize*m_NumSubvectors*sizeof (float ));
107
107
108
108
rType = pq_quantizer->GetReconstructType ();
109
109
CUDA_CHECK (cudaMalloc (&m_DistanceTables, m_BlockSize * m_NumSubvectors * sizeof (float )));
@@ -117,14 +117,14 @@ class GPU_Quantizer {
117
117
m_BlockSize = opq_quantizer->GetBlockSize ();
118
118
m_DimPerSubvector = opq_quantizer->GetDimPerSubvector ();
119
119
120
- LOG (Helper::LogLevel::LL_Debug, " Using OPQ - numSubVectors:%d, KsPerSub:%ld, BlockSize:%ld, DimPerSub:%d, total size of tables:%ld\n " , m_NumSubvectors, m_KsPerSubvector, m_BlockSize, m_DimPerSubvector, m_BlockSize*m_NumSubvectors*sizeof (float ));
120
+ SPTAGLIB_LOG (Helper::LogLevel::LL_Debug, " Using OPQ - numSubVectors:%d, KsPerSub:%ld, BlockSize:%ld, DimPerSub:%d, total size of tables:%ld\n " , m_NumSubvectors, m_KsPerSubvector, m_BlockSize, m_DimPerSubvector, m_BlockSize*m_NumSubvectors*sizeof (float ));
121
121
122
122
rType = opq_quantizer->GetReconstructType ();
123
123
CUDA_CHECK (cudaMalloc (&m_DistanceTables, m_BlockSize * m_NumSubvectors * sizeof (float )));
124
124
CUDA_CHECK (cudaMemcpy (m_DistanceTables, opq_quantizer->GetL2DistanceTables (), m_BlockSize*m_NumSubvectors*sizeof (float ), cudaMemcpyHostToDevice));
125
125
}
126
126
else {
127
- LOG (Helper::LogLevel::LL_Error, " Only PQ and OPQ quantizers are supported for GPU build\n " );
127
+ SPTAGLIB_LOG (Helper::LogLevel::LL_Error, " Only PQ and OPQ quantizers are supported for GPU build\n " );
128
128
exit (1 );
129
129
}
130
130
0 commit comments