Versions Compared

Old Version 9

changes.mady.by.user Anirudh Subramanian

Saved on

compared with

New Version 10

changes.mady.by.user Anirudh Subramanian

Saved on

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

Add a NNVM pass for the backend. This would use the amp lists based on the original_dtype and target_dtype.
This pass will perform graph traversal and add amp_cast and amp_multicast layers for FP16 and FP32 ops based on the op whitelists and excluded_sym_names. Some of the ideas have been borrowed from quantization pass added as part of quantization support [2].

...