diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 04b8bd2f..e5d1e769 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -113,9 +113,8 @@ jobs:
           if ( $env:CUDA_VERSION -eq $env:PYPI_CUDA_VERSION ){
             $env:PYPI_BUILD = 1
           }
-          
-          if (${{ runner.os }} -eq "Linux") { $platform_name = "manylinux2014_x86_64" } else {$platform_name = "win_amd64"}
-          python setup.py sdist bdist_wheel --plat-name $platform_name --python-version ${{ matrix.pyver }}
+
+          python setup.py sdist bdist_wheel
 
       - uses: actions/upload-artifact@v3
         if: runner.os == 'Linux'
@@ -238,13 +237,7 @@ jobs:
           python --version
           which python
 
-          if [[ "${{ runner.os }}" == "Linux" ]]; then
-            platform_name="manylinux2014_x86_64"
-          else
-            platform_name="win_amd64"
-          fi
-          
-          ROCM_VERSION=${{ matrix.rocm }} python setup.py sdist bdist_wheel --plat-name $platform_name --python-version ${{ matrix.python }}
+          ROCM_VERSION=${{ matrix.rocm }} python setup.py sdist bdist_wheel
 
       - uses: actions/upload-artifact@v3
         if: runner.os == 'Linux'
diff --git a/setup.py b/setup.py
index 68969e3c..84b32fc4 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 import requests
 from pathlib import Path
 from setuptools import setup, find_packages
-from torch.utils.cpp_extension import BuildExtension
+from torch.utils.cpp_extension import CUDAExtension
 
 
 def get_latest_kernels_version(repo):
@@ -137,6 +137,11 @@ def get_kernels_whl_url(
         "eval": ["lm_eval>=0.4.0", "tabulate", "protobuf", "evaluate", "scipy"],
         "dev": ["black", "mkdocstrings-python", "mkdocs-material", "griffe-typingdoc"]
     },
-    cmdclass={"build_ext": BuildExtension},
+    ext_modules=[
+        CUDAExtension(
+            name="kernels",
+            sources=[],
+        )
+    ],
     **common_setup_kwargs,
 )