tests/cpp/operator/test_cast_float8blockwise.cu · b944277cdd81a70781e88dcf4d01cfbceee44c5f · OpenDAS / TransformerEngine

[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

test_cast_float8blockwise.cu 25.6 KB

Replace test_cast_float8blockwise.cu