From 9190b0c82c0cbba6ed22616111125f6f39746c2d Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 16 Aug 2024 16:21:17 +0200 Subject: [PATCH] Attempt to remove the specifed compute cap. --- server/custom_kernels/setup.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/server/custom_kernels/setup.py b/server/custom_kernels/setup.py index 69f6b72a..e0b83987 100644 --- a/server/custom_kernels/setup.py +++ b/server/custom_kernels/setup.py @@ -1,10 +1,7 @@ from setuptools import setup from torch.utils.cpp_extension import BuildExtension, CUDAExtension -import torch extra_compile_args = ["-std=c++17"] -if not torch.version.hip: - extra_compile_args.append("-arch=compute_80") setup( name="custom_kernels",