Fix Evoformer compilation when no GPU is present

- this is taken care correctly and more generally by builder.compute_capability_args
- allow for cross-compilation in systems without GPU (see e.g., #5308)

Signed-off-by: Santi Villalba <[email protected]>

Author: sdvillal

op_builder/evoformer_attn.py

@@ -29,18 +29,6 @@ def sources(self):
         src_dir = 'csrc/deepspeed4science/evoformer_attn'
         return [f'{src_dir}/attention.cpp', f'{src_dir}/attention_back.cu', f'{src_dir}/attention_cu.cu']
 
-    def nvcc_args(self):
-        args = super().nvcc_args()
-        try:
-            import torch
-        except ImportError:
-            self.warning("Please install torch if trying to pre-compile kernels")
-            return args
-        major = torch.cuda.get_device_properties(0).major  #ignore-cuda
-        minor = torch.cuda.get_device_properties(0).minor  #ignore-cuda
-        args.append(f"-DGPU_ARCH={major}{minor}")
-        return args
-
     def is_compatible(self, verbose=False):
         try:
             import torch