Skip to content

make torch.set_num_threads also set MKL threads (take 2)#5002

Merged
soumith merged 2 commits intomasterfrom
mklthreads
Feb 2, 2018
Merged

make torch.set_num_threads also set MKL threads (take 2)#5002
soumith merged 2 commits intomasterfrom
mklthreads

Conversation

@soumith
Copy link
Copy Markdown
Collaborator

@soumith soumith commented Feb 1, 2018

Seen #4949 for take 1.

After snooping mkl-service.h and seeing https://software.intel.com/en-us/forums/intel-math-kernel-library/topic/299365 I realized that the correct C prototypes came in camelcase-ish naming. Moved to that.

@ezyang
Copy link
Copy Markdown
Contributor

ezyang commented Feb 2, 2018

LGTM.

@soumith soumith merged commit bf3655a into master Feb 2, 2018
@soumith soumith deleted the mklthreads branch February 2, 2018 14:25
@soumith soumith added the 0.3.1 label Feb 5, 2018
soumith added a commit that referenced this pull request Feb 7, 2018
* torch.set_num_threads sets MKL option too

* fix to use C prototype instead of fortran
laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
* torch.set_num_threads sets MKL option too

* fix to use C prototype instead of fortran
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants