Add more link args

Author: giuliabaldini

build_extension.py

@@ -101,6 +101,7 @@ def build_extensions(self) -> None:
                     extension.extra_compile_args.append(param)
 
                 if platform.system() == "Linux":
+                    extension.extra_link_args.append("-fopenmp")
                     extension.extra_link_args.append("-lgomp")
         try:
             build_ext.build_extensions(self)