mutou0308 commited on
Commit
4f724f2
·
verified ·
1 Parent(s): bc3e084

Delete setup.py

Browse files
Files changed (1) hide show
  1. setup.py +0 -26
setup.py DELETED
@@ -1,26 +0,0 @@
1
- from setuptools import setup
2
- from torch.utils.cpp_extension import BuildExtension, CUDAExtension
3
- import os
4
- import torch
5
-
6
- print("Building gscuda")
7
- # 假设源文件在 gs_cuda 目录下
8
- file_path = "utils/gs_cuda_dmax"
9
-
10
- setup(
11
- name="gscuda", # 模块名
12
- ext_modules=[
13
- CUDAExtension(
14
- name="gscuda", # 可以直接作为模块导入
15
- sources=[
16
- os.path.join(file_path, "gswrapper.cpp"),
17
- os.path.join(file_path, "gs.cu")
18
- ],
19
- # 设置运行时库路径(可选)
20
- library_dirs=[os.path.join(os.path.dirname(torch.__file__), 'lib')],
21
- )
22
- ],
23
- cmdclass={
24
- "build_ext": BuildExtension
25
- },
26
- )