mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
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.
16 lines
359 B
16 lines
359 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, |
|
)
|
|
|