add cuda support for `plmblast.py`
Opened this issue · 0 comments
MiTRonGTE commented
It seems that part of mathematical operations should be done on gpu - which may speed up the whole search.
This task will require to create a specific embedding loader for main loop. Which should
- load embeddings directly into cuda device ( use
map_location
intorch.load
) - calculate substitution matrix and optionally signal enh
both ongpu
- remove the need of
concurrent/multiprocessing
packages all multiprocessing routines should usetorch.multiprocessing
Further main loop should work on cpu, (from this point:
Line 62 in 24c136d
For now I suggest to work on copy of scripts/plmblast.py
for instance scripts/plmblastgpu.py