[cuDNN V8 API] (reopen 2) Allow the number of kernels profiled under torch.backends.cudnn.benchmark = True to be limitedCudnnv8 benchmark limit#78299
Conversation
🔗 Helpful links
✅ No Failures (0 Pending)As of commit 58b21c2 (more details on the Dr. CI page): Expand to see more💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
42ed219 to
6e15bcd
Compare
ecc5742 to
58b21c2
Compare
|
@ngimel has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@pytorchbot merge |
|
@pytorchbot successfully started a merge job. Check the current status here |
|
Hey @eqy. |
…torch.backends.cudnn.benchmark = True to be limitedCudnnv8 benchmark limit (#78299) (#78299) Summary: Reopen of #77002 to address comments by malfet CC ngimel ptrblck Pull Request resolved: #78299 Approved by: https://github.com/ngimel Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/ae6dd20ba7725ea7c10d759f82891d86eb724c11 Reviewed By: mehtanirav, cpuhrsch Differential Revision: D37690491 fbshipit-source-id: 6c603a706663dcc7086755a5d4b6e382233dad7b
Reopen of #77002 to address comments by @malfet
CC @ngimel @ptrblck