Skip to content

Commit a66ad0c

Browse files
committed
Allow easy-search to work with --gpu 1
1 parent 81171a5 commit a66ad0c

File tree

2 files changed

+19
-0
lines changed

2 files changed

+19
-0
lines changed

data/workflow/easysearch.sh

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,15 @@ if notExists "${TARGET}.dbtype"; then
2121
|| fail "target createdb died"
2222
fi
2323
TARGET="${TMP_PATH}/target"
24+
25+
if [ -n "${GPU}" ]; then
26+
if notExists "${TMP_PATH}/target_pad"; then
27+
# shellcheck disable=SC2086
28+
"$MMSEQS" makepaddedseqdb "${TMP_PATH}/target" "${TMP_PATH}/target_pad" ${MAKEPADDEDSEQDB_PAR} \
29+
|| fail "makepaddedseqdb died"
30+
fi
31+
TARGET="${TMP_PATH}/target_pad"
32+
fi
2433
fi
2534

2635
if [ -n "${LINSEARCH}" ] && notExists "${TARGET}.linidx"; then
@@ -66,6 +75,14 @@ if [ -n "${REMOVE_TMP}" ]; then
6675
# shellcheck disable=SC2086
6776
"$MMSEQS" rmdb "${TMP_PATH}/target_h" ${VERBOSITY}
6877
fi
78+
79+
if [ -f "${TMP_PATH}/target_pad" ]; then
80+
# shellcheck disable=SC2086
81+
"$MMSEQS" rmdb "${TMP_PATH}/target_pad" ${VERBOSITY}
82+
# shellcheck disable=SC2086
83+
"$MMSEQS" rmdb "${TMP_PATH}/target_pad_h" ${VERBOSITY}
84+
fi
85+
6986
# shellcheck disable=SC2086
7087
"$MMSEQS" rmdb "${TMP_PATH}/query" ${VERBOSITY}
7188
# shellcheck disable=SC2086

src/workflow/EasySearch.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,8 @@ int doeasysearch(int argc, const char **argv, const Command &command, bool linse
138138
par.shuffleDatabase = origShuffle;
139139
par.createdbMode = Parameters::SEQUENCE_SPLIT_MODE_HARD;
140140
cmd.addVariable("CREATEDB_PAR", par.createParameterString(par.createdb).c_str());
141+
cmd.addVariable("GPU", par.gpu ? "TRUE" : NULL);
142+
cmd.addVariable("MAKEPADDEDSEQDB_PAR", par.createParameterString(par.makepaddedseqdb).c_str());
141143
cmd.addVariable("CONVERT_PAR", par.createParameterString(par.convertalignments).c_str());
142144
cmd.addVariable("SUMMARIZE_PAR", par.createParameterString(par.summarizeresult).c_str());
143145

0 commit comments

Comments
 (0)