Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add cuda support for plmblast.py #36

Open
3 tasks
MiTRonGTE opened this issue Jan 5, 2024 · 0 comments
Open
3 tasks

add cuda support for plmblast.py #36

MiTRonGTE opened this issue Jan 5, 2024 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@MiTRonGTE
Copy link
Collaborator

MiTRonGTE commented Jan 5, 2024

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 in torch.load)
  • calculate substitution matrix and optionally signal enh
    both on gpu
  • remove the need of concurrent/multiprocessing packages all multiprocessing routines should use torch.multiprocessing

Further main loop should work on cpu, (from this point:

def submatrix_to_span(self, densitymap, mode: str = 'results') -> pd.DataFrame:
) it will be the best to cpu and gpu based steps will be async so there will be on bottlenecks

For now I suggest to work on copy of scripts/plmblast.py for instance scripts/plmblastgpu.py

@MiTRonGTE MiTRonGTE self-assigned this Jan 5, 2024
@Argusmocny Argusmocny added the enhancement New feature or request label Jul 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
Status: In Progress
Development

When branches are created from issues, their pull requests are automatically linked.

2 participants