"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "ae43f09ef72683c044a635cdedbe67329583e2bc"
Commit 80caf79d authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Prevent excessive parallelism in PyTorch.

We're already using as many processes in parallel as we have CPU cores.
Furthermore, the number of core may be incorrectly calculated as 36
(we've seen this in pytest-xdist) which make compound the problem.

PyTorch performance craters without this.
parent bb3bfa2d
...@@ -4,6 +4,8 @@ jobs: ...@@ -4,6 +4,8 @@ jobs:
working_directory: ~/transformers working_directory: ~/transformers
docker: docker:
- image: circleci/python:3.5 - image: circleci/python:3.5
environment:
OMP_NUM_THREADS: 1
resource_class: xlarge resource_class: xlarge
parallelism: 1 parallelism: 1
steps: steps:
...@@ -19,6 +21,8 @@ jobs: ...@@ -19,6 +21,8 @@ jobs:
working_directory: ~/transformers working_directory: ~/transformers
docker: docker:
- image: circleci/python:3.5 - image: circleci/python:3.5
environment:
OMP_NUM_THREADS: 1
resource_class: xlarge resource_class: xlarge
parallelism: 1 parallelism: 1
steps: steps:
...@@ -34,6 +38,8 @@ jobs: ...@@ -34,6 +38,8 @@ jobs:
working_directory: ~/transformers working_directory: ~/transformers
docker: docker:
- image: circleci/python:3.5 - image: circleci/python:3.5
environment:
OMP_NUM_THREADS: 1
resource_class: xlarge resource_class: xlarge
parallelism: 1 parallelism: 1
steps: steps:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment