KellenSunderland 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_r240106765
##########
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:
We should be able to run this test on CPU in CI. Could we test to see if
'MKL' is in the env var 'BUILD_TAG' and run the test if it is.
----------------------------------------------------------------
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
