Code

Offloading example#1299

Code

Open

sanandaraj5597 wants to merge 8 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom sanandaraj5597:offloading_examplesanandaraj5597/TransformerEngine:offloading_exampleCopy head branch name to clipboard

+32

Commits

Commits on Aug 23, 2024

• Added offloading support FP8 attention

  Show description for 0ef5803
  

  Selvaraj Anandaraj

  committed
  0ef5803

Commits on Sep 4, 2024

• Merge branch 'main' into main

  ksivaman

  authored
  84857bb

• Update transformer_engine/pytorch/attention.py

  Show description for e59e4ba
  

  sanandaraj5597

  and

  ksivaman

  authored
  e59e4ba

Commits on Sep 5, 2024

• Fix

  Show description for e54c03b
  ksivaman

  committed
  e54c03b

• Merge branch 'main' into main

  Show description for 4e249fb
  ksivaman

  authored
  4e249fb

Commits on Oct 29, 2024

• Merge branch 'NVIDIA:main' into main

  sanandaraj5597

  authored
  7348d21

• Added example for CPU offloading

  Show description for 69866e2
  

  Selvaraj Anandaraj

  committed
  69866e2

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for dc89e7d
  pre-commit-ci[bot]

  committed
  dc89e7d