Skip to content

GitLab

Commit b85ee5c1 authored by Casper's avatar Casper
Browse files

Move kernels to base directory

parent 1e82667b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
setup.py
View file @ b85ee5c1
...@@ -26,10 +26,10 @@ if build_cuda_extension: ...@@ -26,10 +26,10 @@ if build_cuda_extension:
CUDAExtension( CUDAExtension(
name="awq_inference_engine", name="awq_inference_engine",
sources=[ sources=[
"awq/kernels/csrc/pybind.cpp", "awq_cuda/pybind.cpp",
"awq/kernels/csrc/quantization/gemm_cuda_gen.cu", "awq_cuda/quantization/gemm_cuda_gen.cu",
"awq/kernels/csrc/layernorm/layernorm.cu", "awq_cuda/layernorm/layernorm.cu",
"awq/kernels/csrc/position_embedding/pos_encoding_kernels.cu" "awq_cuda/position_embedding/pos_encoding_kernels.cu"
], ],
extra_compile_args={ extra_compile_args={
"cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"], "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment