search

pytorch / builder

Continuous builder and binary build scripts for pytorch
BSD 2-Clause "Simplified" License
323 stars 213 forks source link

Add `OVERRIDE_GENCODE_CUDNN` to allow overriding CUDNN gencode #1892

Closed xuzhao9 closed 1 week ago

xuzhao9 commented 1 week ago

This PR adds an option to override the CUDNN gencode so that the downstream docker build scripts can include sm_90a arch, which is needed for benchmarking on H100.

atalman commented 1 week ago

@xuzhao9 this will rebuild docker images. We probably need to commit this to a branch, and try to build these with special tags. Then try to use these tags in the test PR against pytorch/pytorch and torchbench

atalman commented 1 week ago

cc @ptrblck @nWEIdia