Browse Source

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

GitOrigin-RevId: b2dbda96be
release-1.1
Megvii Engine Team 4 years ago
parent
commit
a9f98e9c66
3 changed files with 5 additions and 8 deletions
  1. +2
    -2
      dnn/scripts/gen_param_defs.py
  2. +0
    -3
      imperative/.gitignore
  3. +3
    -3
      imperative/python/setup.py

+ 2
- 2
dnn/scripts/gen_param_defs.py View File

@@ -345,14 +345,14 @@ class PyWriter(IndentWriterBase):
if not self._imperative: if not self._imperative:
self._write( self._write(
'def _as_dtype_num(dtype):\n' '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' ' return m._get_dtype_num(dtype)\n'
'\n' '\n'
) )


self._write( self._write(
'def _as_serialized_dtype(dtype):\n' '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' ' return m._get_serialized_dtype(dtype)\n'
'\n' '\n'
) )


+ 0
- 3
imperative/.gitignore View File

@@ -1,5 +1,2 @@
Makefile Makefile
/test/imperative_test /test/imperative_test
*.so
/python/megengine/core/ops/_internal/generated_ops.py
/python/megengine/core/ops/_internal/param_defs.py

+ 3
- 3
imperative/python/setup.py View File

@@ -52,12 +52,12 @@ if local_version:
__version__ = '{}+{}'.format(__version__, local_version) __version__ = '{}+{}'.format(__version__, local_version)


packages = find_packages(exclude=['test']) packages = find_packages(exclude=['test'])
package_data = [
megengine_data = [
str(f.relative_to('megengine')) str(f.relative_to('megengine'))
for f in pathlib.Path('megengine', 'core', 'include').glob('**/*') for f in pathlib.Path('megengine', 'core', 'include').glob('**/*')
] ]


package_data += [
megengine_data += [
str(f.relative_to('megengine')) str(f.relative_to('megengine'))
for f in pathlib.Path('megengine', 'core', 'lib').glob('**/*') for f in pathlib.Path('megengine', 'core', 'lib').glob('**/*')
] ]
@@ -80,7 +80,7 @@ setup_kwargs = dict(
author_email=email, author_email=email,
packages=packages, packages=packages,
package_data={ package_data={
'megengine': package_data,
'megengine': megengine_data,
}, },
ext_modules=prebuild_modules, ext_modules=prebuild_modules,
install_requires=requires, install_requires=requires,


Loading…
Cancel
Save