Thuyen Ngo e35bf56461 [Bazel] Add CUDA build to CI (#66241) ...

Summary:
Fixes https://github.com/pytorch/pytorch/issues/35316
On master, bazel cuda build is disabled due to lack of a proper `cu_library` rule. This PR:
- Add `rules_cuda` to the WORKSPACE and forward `cu_library` to `rules_cuda`.
- Use a simple local cuda and cudnn repositories (adopted from TRTorch) for cuda 11.3.
- Fix current broken cuda build.
- Enable cuda build in CI, not just for `:torch` target but all the test binaries to catch undefined symbols.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/66241

Reviewed By: ejguan

Differential Revision: D31544091

Pulled By: malfet

fbshipit-source-id: fd3c34d0e8f80fee06f015694a4c13a8e9e12206

2021-12-17 13:44:29 -08:00

..

BUILD

remediation of S205607

2020-07-17 17:19:47 -07:00

cu.bzl

[Bazel] Add CUDA build to CI (#66241)

2021-12-17 13:44:29 -08:00

workspace.bzl

[codemod][fbcode/caffe2] Apply all buildifier fixes

2021-09-23 14:03:19 -07:00