mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
377 B
19 lines
377 B
import os
|
|
|
|
from setuptools import Extension, setup
|
|
|
|
this_dir = os.path.dirname(os.path.abspath(__file__))
|
|
ext_modules = [
|
|
Extension(
|
|
"rotorc",
|
|
sources=[os.path.join(this_dir, "ckpt_solver_rotor.c")],
|
|
)
|
|
]
|
|
|
|
setup(
|
|
name="rotor c extension",
|
|
version="0.1",
|
|
description="rotor c extension for faster dp computing",
|
|
ext_modules=ext_modules,
|
|
)
|