Skip to content

support OMP_NUM_THREADS for CPU limit #136

Description

@alexweisberg

Hi, I am looking into an issue with a separate program that uses tape (zhangyumeng1sjtu/T4SEfinder#2). I am running this program on a computing cluster that does not have a GPU, and so I need to limit the number of threads spawned by each program to a specific number. The program I am using is called T4SEfinder, which calls the transformer function of tape as follows:
tape-embed transformer {fasta} {embed_path} {pretrained_model} --tokenizer iupac --batch_size 1 --seed {random_seed} --log_level WARNING
and that is spawning as many processes as the machine has CPUs.

Would it be possible for tape to respect the OMP_NUM_THREADS environment variable when running? Or alternatively adding a number of threads argument? Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions