Logo TT-NN

TTNN

  • What is ttnn?
  • Getting Started
  • Using ttnn
  • Tensor
  • APIs
  • Tutorials
    • Tensor and Add Operation
    • Matmul Operation
    • Multi-Head Attention
    • ttnn Tracer
    • ttnn Profiling
    • Resnet Basic Block
    • Graphing Torch DiT_XL_2 With TTNN
      • Build a graph of a pytorch based model
  • Onboarding New Functionality
  • Converting torch Model to ttnn
  • Adding New ttnn Operation
  • Profiling ttnn Operations
  • Dependencies
  • Building and Uplifting Demos
  • Placeholder title

Models

  • Getting Started
  • Performance

Resources

  • Support
  • Contributing as a developer
TT-NN
  • Tutorials
  • Graphing Torch DiT_XL_2 With TTNN
  • View page source

Graphing Torch DiT_XL_2 With TTNN

Note

TT-NN tutorials currently work on Grayskull only.

  • Build a graph of a pytorch based model
    • Clone the library from https://github.com/facebookresearch/DiT.git
    • Download DiT-XL/2 Models
    • Sample from Pre-trained DiT Models and build the graph
    • Display the graph
Previous Next

© Copyright Tenstorrent.

Built with Sphinx using a theme provided by Read the Docs.
Version: v0.49.0
Versions
latest
v0.49.0