From f1280e9e0aa1d8ad3f9815a8033e7cb2dd7ad5fb Mon Sep 17 00:00:00 2001
From: Stas Bekman <stas00@users.noreply.github.com>
Date: Fri, 4 Dec 2020 22:57:40 -0800
Subject: [PATCH] [build] add compute_86

RTX-30 series are compute_86
```
python -c "import torch; print(torch.cuda.get_device_capability())"
```
This PR adds support for this compute capability.

Reference: https://developer.nvidia.com/cuda-gpus
---
 op_builder/builder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/op_builder/builder.py b/op_builder/builder.py
index ccbcd9aaf5ec..6e202068bd93 100644
--- a/op_builder/builder.py
+++ b/op_builder/builder.py
@@ -34,7 +34,7 @@ def installed_cuda_version():
 def get_default_compute_capatabilities():
     compute_caps = DEFAULT_COMPUTE_CAPABILITIES
     if installed_cuda_version()[0] >= 11:
-        compute_caps += ";8.0"
+        compute_caps += ";8.0;8.6"
     return compute_caps