Skip to content

NNFusion v0.1 Release

Compare
Choose a tag to compare
@wenxcs wenxcs released this 05 Nov 04:42
· 155 commits to master since this release
8ae20c6
  • Build and Installation:

    • Support out-of-box installation with docker image
    • Support source code install on native system and docker
    • Support devices like CUDA GPUs, and ROCm GPUs.
  • Models, Framework and Operators:

    • Support DNN model formats including TensorFlow and ONNX
    • Support commonly used models including AlexNet, VGG11, ResNet50, seq2seq, BERT, etc.
    • Support more than 100 commonly used operators.
  • Model Compilation and Execution:

    • Provide a full-stack optimization mechanism, including data-flow graph optimizations, model-specific kernel selection, kernel co-scheduling, etc.
    • Provide ahead-of-time and source-to-source(model-to-code) compilation to reduce runtime overhead
    • Remove third-party library or framework dependencies
  • Usability:

    • Provide command line tool nnfusion
    • Provide tools for users to freeze TensorFlow and PyTorch models
    • Provide flexible way to customize optimization through direct code modification on generated code

中文版本说明快捷通道--> #72 (comment)