Error if GPU modules are used with non-gpu dbs

milot-mirdita · milot-mirdita · commit 81171a53dee9 · 2024-11-13T13:46:19.000+09:00
diff --git a/src/commons/Parameters.h b/src/commons/Parameters.h
@@ -90,6 +90,7 @@ class Parameters {
     static const unsigned int DBTYPE_EXTENDED_COMPRESSED = 1;
     static const unsigned int DBTYPE_EXTENDED_INDEX_NEED_SRC = 2;
     static const unsigned int DBTYPE_EXTENDED_CONTEXT_PSEUDO_COUNTS = 4;
+    static const unsigned int DBTYPE_EXTENDED_GPU = 8;
 
     // don't forget to add new database types to DBReader::getDbTypeName and Parameters::PARAM_OUTPUT_DBTYPE
 
diff --git a/src/prefiltering/ungappedprefilter.cpp b/src/prefiltering/ungappedprefilter.cpp
@@ -391,6 +391,15 @@ int prefilterInternal(int argc, const char **argv, const Command &command, int m
     IndexReader * qDbrIdx = NULL;
     DBReader<unsigned int> * qdbr = NULL;
     DBReader<unsigned int> * tdbr = tDbrIdx.sequenceReader;
+
+    if (par.gpu == true) {
+        const bool isGpuDb = DBReader<unsigned int>::getExtendedDbtype(tdbr->getDbtype()) & Parameters::DBTYPE_EXTENDED_GPU;
+        if (isGpuDb == false) {
+            Debug(Debug::ERROR) << "Database " << FileUtil::baseName(par.db2) << " is not a valid GPU database\n";
+            EXIT(EXIT_FAILURE);
+        }
+    }
+
     const int targetSeqType = tdbr->getDbtype();
     int querySeqType;
     if (sameDB == true) {
diff --git a/src/util/gpuserver.cpp b/src/util/gpuserver.cpp
@@ -29,6 +29,12 @@ int gpuserver(int argc, const char **argv, const Command& command) {
     IndexReader dbrIdx(par.db1, par.threads, IndexReader::SEQUENCES, (touch) ? (IndexReader::PRELOAD_INDEX | IndexReader::PRELOAD_DATA) : 0 );
     DBReader<unsigned int>* dbr = dbrIdx.sequenceReader;
 
+    const bool isGpuDb = DBReader<unsigned int>::getExtendedDbtype(dbr->getDbtype()) & Parameters::DBTYPE_EXTENDED_GPU;
+    if (isGpuDb == false) {
+        Debug(Debug::ERROR) << "Database " << FileUtil::baseName(par.db1) << " is not a valid GPU database\n";
+        EXIT(EXIT_FAILURE);
+    }
+
     std::vector<size_t> offsets;
     offsets.reserve(dbr->getSize() + 1);
 
diff --git a/src/util/makepaddedseqdb.cpp b/src/util/makepaddedseqdb.cpp
@@ -23,9 +23,10 @@ int makepaddedseqdb(int argc, const char **argv, const Command &command) {
 
     SubstitutionMatrix subMat(par.scoringMatrixFile.values.aminoacid().c_str(), 2.0, par.scoreBias);
 
-    DBWriter dbsw(par.db2.c_str(), par.db2Index.c_str(), par.threads, false, dbr.getDbtype());
+    int dbType = DBReader<unsigned int>::setExtendedDbtype(dbr.getDbtype(), Parameters::DBTYPE_EXTENDED_GPU);
+    DBWriter dbsw(par.db2.c_str(), par.db2Index.c_str(), par.threads, false, dbType);
     dbsw.open();
-    DBWriter dbhw(par.hdr2.c_str(), par.hdr2Index.c_str(), par.threads, false, dbr.getDbtype());
+    DBWriter dbhw(par.hdr2.c_str(), par.hdr2Index.c_str(), par.threads, false, Parameters::DBTYPE_GENERIC_DB);
     dbhw.open();
 
     // need to prune low scoring k-mers through masking
