tests/pytorch/references/blockwise_quantizer_reference.py · b944277cdd81a70781e88dcf4d01cfbceee44c5f · OpenDAS / TransformerEngine

"backend/vscode:/vscode.git/clone" did not exist on "ea6b8984abfde2ae0cde8268ae1a3276bd650f35"

[Blockwise] Add support block_len=64 support · b944277c

wenjh authored Jun 25, 2025



Add env to chose blocklen of blockwise quantize.
Signed-off-by: wenjh <wenjh@sugon.com>

Fix pytest of blockwise error
Signed-off-by: wenjh <wenjh@sugon.com>

Resolve new api in  int8 gemm test
Signed-off-by: wenjh <wenjh@sugon.com>

Fix incorrect launch parm
Signed-off-by: wenjh <wenjh@sugon.com>

Fix 1D blockwise(64) acc error
Signed-off-by: wenjh <wenjh@sugon.com>

b944277c

blockwise_quantizer_reference.py 11.1 KB

Replace blockwise_quantizer_reference.py