Skip to content

Enable resource_var when XLA is on for run_squad.py. - #539

Merged
swethmandava merged 2 commits into
NVIDIA:masterfrom
trentlo:enable_res_var
Jun 10, 2020
Merged

Enable resource_var when XLA is on for run_squad.py.#539
swethmandava merged 2 commits into
NVIDIA:masterfrom
trentlo:enable_res_var

Conversation

@trentlo

@trentlo trentlo commented May 29, 2020

Copy link
Copy Markdown
Contributor

I observed 9% perf gain on 8GPUs and 13% on 1GPU for BERT SQuad fine-tuning (fp16, BS=3, on DGX-1), excluding compilation time.

@trentlo

trentlo commented Jun 9, 2020

Copy link
Copy Markdown
Contributor Author

@swethmandava: please help to take a look.

@swethmandava
swethmandava merged commit 9f446c5 into NVIDIA:master Jun 10, 2020
PeganovAnton pushed a commit to PeganovAnton/DeepLearningExamples that referenced this pull request Sep 8, 2020
Enable resource_var when XLA is on for run_squad.py.
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