mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-28 11:45:23 +00:00
1df98d5b66
* [autoparallel] add rotor c version * [fx] remove metainfoprop in rotor solver * [autoparallel] modify C code format * [autoparallel] remove build.py * [autoparallel] fix C extension build * [autoparallel] add C solver consistency test * [autoparallel] remove some unused imports * [autoparallel] refactor rotor solver code * [autoparallel] replace print with colossalai logger * [autoparallel] ranks fixed
16 lines
377 B
Python
16 lines
377 B
Python
from setuptools import setup, Extension
|
|
import os
|
|
|
|
this_dir = os.path.dirname(os.path.abspath(__file__))
|
|
ext_modules = [Extension(
|
|
'dynamic_programs_C_version',
|
|
sources=[os.path.join(this_dir, 'dynamic_programs.c')],
|
|
)]
|
|
|
|
setup(
|
|
name='rotor c extension',
|
|
version='0.1',
|
|
description='rotor c extension for faster dp computing',
|
|
ext_modules=ext_modules,
|
|
)
|