ext_utils

class CMakeExtension(name, source_dir=None)[源代码]

基类:setuptools.extension.Extension

build_with_command(ext_builder)[源代码]

Custom build_ext.build_extension in Extension instead of Command. ext_builder is the instance of build_ext command.

class FasterTransformerExtension(name, source_dir=None)[源代码]

基类:paddlenlp.ops.ext_utils.CMakeExtension

build_with_command(ext_builder)[源代码]

Custom build_ext.build_extension in Extension instead of Command. ext_builder is the instance of build_ext command.

class BuildExtension(*args, **kwargs)[源代码]

基类:paddle.utils.cpp_extension.cpp_extension.BuildExtension

Support both CppExtention of Paddle and custom extensions of PaddleNLP.