Skip to content

GitLab

  • Menu
Projects Groups Snippets
    • Loading...
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in / Register
  • T TransformerEngine
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 0
    • Issues 0
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 0
    • Merge requests 0
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Monitor
    • Monitor
    • Incidents
  • Packages & Registries
    • Packages & Registries
    • Package Registry
    • Infrastructure Registry
  • Analytics
    • Analytics
    • CI/CD
    • Repository
    • Value stream
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • OpenDAS
  • TransformerEngine
  • Repository

Switch branch/tag
  • TransformerEngine
  • tests
  • libcustom_ort_fp8_qdq_ops.so
Find file HistoryPermalink
  • asfiyab-nvidia's avatar
    Add ONNX export support for TE modules (#41) · 6c9ce179
    asfiyab-nvidia authored Jan 18, 2023
    
    
    * Add ONNX export support for TE modules (#1)
    
    * Add TorchScript Operators
    * Add symbolic methods to ONNX exporter
    * Add tests for the ONNX export
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fixes for pylint tests
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fix pylint warning in softmax.py
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * move FP8 ORT lib inside tests/
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * enable cross attention tests
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * refactor code by @nzmora
    * Increase layernorm FP16 threshold
    * Normalize onnx file names: _ separates configs; - separates words in a single config
    * Add get_attn_mask_str and fix mask string
    * Add missing ONNX files
    * Moved generated ONNX files to tests/gen_onnx_models/
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fix merge conflict changes
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fix Q/DQ scale input
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * enable FP16 config when bias is disabled
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fix pylint check errors
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * updates
    1. remove List import for pylint failure
    2. address comments: remove state tensors from GPU
    3. address comments: Update reverse_map_dtype function and add to namespace
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * minor fix: coding guidelines
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * changes:
    1. skip FP8 tests on  non-hopper devices
    2. minor fix for C++ lint check
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * fix onnxruntime version
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * minor fix: add space between code and comment
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * changes
    1. update copyrights
    2. update path to ORT .so
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    
    * Apply suggestions from code review
    Co-authored-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
    Signed-off-by: default avatarasfiyab-nvidia <117682710+asfiyab-nvidia@users.noreply.github.com>
    Signed-off-by: default avatarAsfiya Baig <asfiyab@nvidia.com>
    Signed-off-by: default avatarasfiyab-nvidia <117682710+asfiyab-nvidia@users.noreply.github.com>
    Co-authored-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
    6c9ce179
libcustom_ort_fp8_qdq_ops.so 660 KB

Download (660 KB)

Replace libcustom_ort_fp8_qdq_ops.so

Attach a file by drag & drop or click to upload


Cancel
A new branch will be created in your fork and a new merge request will be started.