Skip to content

Add benchmarks for thread tuning.#6283

Merged
tfboyd merged 3 commits intomasterfrom
yuefengz_resnet_keras
Feb 28, 2019
Merged

Add benchmarks for thread tuning.#6283
tfboyd merged 3 commits intomasterfrom
yuefengz_resnet_keras

Conversation

@yuefengz
Copy link
Contributor

Also removed the bfc allocator one. We'll add GPU thread pool, inter op, intra op pool later.

@yuefengz yuefengz requested a review from tfboyd February 28, 2019 09:12
@yuefengz yuefengz requested review from a team and karmel as code owners February 28, 2019 09:12
Copy link
Member

@tfboyd tfboyd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just one change please.


def benchmark_8_gpu_bfc_allocator(self):
"""Restricts CPU memory allocation."""
def benchmark_8_gpu_tweaked(self):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you go ahead and delete this test completely and make the test above def benchmark_8_gpu(self): run with the 14 threads. Since the goal for this class of tests is to test accuracy first it is ok for the accuracy test to be "tuned". The short test are our primary location for perf tests. I realize this is a bit weird. If the accuracy tests did not take 16 hours I would have a normal and then tweaked version. Thank you and this is a great test to track our progress.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.

Copy link
Member

@tfboyd tfboyd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, you rock. Not having to add all the tests myself is awesome. Seems silly but I us to add and manage them. Team work is so much better.

@tfboyd tfboyd merged commit 54dffe2 into master Feb 28, 2019
@yuefengz yuefengz deleted the yuefengz_resnet_keras branch June 13, 2019 23:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants