--cpu-threads failes when calling 'corr'

kordk / torch-ecpg

(GPU accelerated) eCpG mapper

BSD 3-Clause "New" or "Revised" License

2 stars 0 forks source link

--cpu-threads failes when calling 'corr' #4

Closed kordk closed 1 year ago

kordk commented 1 year ago

When I attempt to set the cpu thread limit with corr, it is not an available option:

tecpg run corr --cpu-threads 1
Usage: tecpg run corr [OPTIONS]
Try 'tecpg run corr --help' for help.

Error: No such option: --cpu-threads

liamgd commented 1 year ago

The --cpu-threads flag is placed directly after "tecpg" like so: tecpg --cpu-threads [CPU THREADS] run corr. This is done because other commands also use pytorch, and having the flag modify the root "tecpg" group allows for other commands besides corr to have the correct number of CPU threads with the same flag.