Update for build.toml changes

build.toml

@@ -1,17 +1,14 @@
 [general]
-version = "0.0.1"
+name = "quantization_eetq"
 
 [torch]
-name = "quantization_eetq"
 src = [
-  "torch-ext/registration.h",
   "torch-ext/torch_binding.cpp",
   "torch-ext/torch_binding.h"
 ]
-pyroot = "torch-ext"
 
 [kernel.cutlass_kernels]
-capabilities = [ "7.0", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
+cuda-capabilities = [ "7.0", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
 src = [
   "cutlass_extensions/include/cutlass_extensions/arch/mma.h",
   "cutlass_extensions/include/cutlass_extensions/compute_occupancy.h",
@@ -61,7 +58,7 @@ depends = [ "cutlass_2_10", "torch" ]
 include = [ ".", "utils", "cutlass_extensions/include" ]
 
 [kernel.weight_only_batched_gemv]
-capabilities = [ "7.0", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
+cuda-capabilities = [ "7.0", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
 src = [
   "cutlass_extensions/include/cutlass_extensions/interleaved_numeric_conversion.h",
   "cutlass_extensions/include/cutlass_extensions/gemm/kernel/mixed_gemm_B_layout.h",

torch-ext/registration.h

@@ -1,27 +0,0 @@
-#pragma once
-
-#include <Python.h>
-
-#define _CONCAT(A, B) A##B
-#define CONCAT(A, B) _CONCAT(A, B)
-
-#define _STRINGIFY(A) #A
-#define STRINGIFY(A) _STRINGIFY(A)
-
-// A version of the TORCH_LIBRARY macro that expands the NAME, i.e. so NAME
-// could be a macro instead of a literal token.
-#define TORCH_LIBRARY_EXPAND(NAME, MODULE) TORCH_LIBRARY(NAME, MODULE)
-
-// A version of the TORCH_LIBRARY_IMPL macro that expands the NAME, i.e. so NAME
-// could be a macro instead of a literal token.
-#define TORCH_LIBRARY_IMPL_EXPAND(NAME, DEVICE, MODULE) \
-  TORCH_LIBRARY_IMPL(NAME, DEVICE, MODULE)
-
-// REGISTER_EXTENSION allows the shared library to be loaded and initialized
-// via python's import statement.
-#define REGISTER_EXTENSION(NAME)                                               \
-  PyMODINIT_FUNC CONCAT(PyInit_, NAME)() {                                     \
-    static struct PyModuleDef module = {PyModuleDef_HEAD_INIT,                 \
-                                        STRINGIFY(NAME), nullptr, 0, nullptr}; \
-    return PyModule_Create(&module);                                           \
-  }