提交 a9f98e9c 编写于 作者: M Megvii Engine Team

refactor(meg/internal): move interal codes back to megbrain

GitOrigin-RevId: b2dbda96be73e27733bada2858011d2c08b221ff
上级 65c54913
......@@ -345,14 +345,14 @@ class PyWriter(IndentWriterBase):
if not self._imperative:
self._write(
'def _as_dtype_num(dtype):\n'
' import megengine._internal.mgb as m\n'
' import megbrain.mgb as m\n'
' return m._get_dtype_num(dtype)\n'
'\n'
)
self._write(
'def _as_serialized_dtype(dtype):\n'
' import megengine._internal.mgb as m\n'
' import megbrain.mgb as m\n'
' return m._get_serialized_dtype(dtype)\n'
'\n'
)
......
Makefile
/test/imperative_test
*.so
/python/megengine/core/ops/_internal/generated_ops.py
/python/megengine/core/ops/_internal/param_defs.py
......@@ -52,12 +52,12 @@ if local_version:
__version__ = '{}+{}'.format(__version__, local_version)
packages = find_packages(exclude=['test'])
package_data = [
megengine_data = [
str(f.relative_to('megengine'))
for f in pathlib.Path('megengine', 'core', 'include').glob('**/*')
]
package_data += [
megengine_data += [
str(f.relative_to('megengine'))
for f in pathlib.Path('megengine', 'core', 'lib').glob('**/*')
]
......@@ -80,7 +80,7 @@ setup_kwargs = dict(
author_email=email,
packages=packages,
package_data={
'megengine': package_data,
'megengine': megengine_data,
},
ext_modules=prebuild_modules,
install_requires=requires,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册