DeepSpeed
591744eb - Support N-dimension input in quantization kernel (#2575)

Commit
2 years ago
Support N-dimension input in quantization kernel (#2575) * Add support for inputs > 2D * use vec * Add N-Dim support to Dequant kernel * merge master and fix format * Bug Fix * fix num_bits * Fix dequant Co-authored-by: Connor Holmes <connorholmes@microsoft.com>
Author
Parents
  • csrc/quantization
    • File
      pt_binding.cpp