mirror of https://github.com/hpcaitech/ColossalAI
17 lines
359 B
Python
17 lines
359 B
Python
|
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,
|
||
|
|
)
|