pengzhao-intel commented on a change in pull request #12922: Support Quantized
Fully Connected by INT8 GEMM
URL: https://github.com/apache/incubator-mxnet/pull/12922#discussion_r240112241
##########
File path: tests/python/quantization/test_quantization.py
##########
@@ -270,7 +270,7 @@ def check_quantized_pooling(data_shape, kernel, pool_type,
pad, stride, global_p
def test_quantized_fc():
def check_quantized_fc(data_shape, num_hidden, no_bias, qdtype,
flatten=True):
if mx.current_context().device_type != 'gpu':
Review comment:
@KellenSunderland good suggestion! Currently, the CI doesn't include Intel
MKL library as BLAS library and @azai91 is working on adding it so that we can
have a better coverage, such as batch_gemm, quantization FC, etc.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
