fix: update linker flag for OpenMP support in setup.py

setup.py

@@ -17,7 +17,7 @@
 # check if openmp is available
 if torch.backends.openmp.is_available():
     extra_compile_args["cxx"] = ["-fopenmp"]
-    extra_link_args.append("-lgomp")
+    extra_link_args.append("-fopenmp")
 
 setuptools.setup(
     name=NAME,