-
Notifications
You must be signed in to change notification settings - Fork 6.8k
MKLDNN-1.0 doesn't support slice operator for Large Tensor #16732
Comments
@mxnet-label-bot add [MKLDNN] |
@rongzha1 @wuxun-zhang could you take a look ASAP? |
@access2rohit is this a necessary part for r1.6 or we can fix in master? |
@pengzhao-intel I am looking into this. |
Can't reproduce this case in our skylake machine . Will keep debug |
@access2rohit could you add some bt info? thanks |
@rongzha1 Have you ever tried to build MXNet with |
@TaoLv Yes, we have enabled the int64 flag, building command is I also cannot reproduce this issue. However, I found another bug about the offset assignment in slice and will file a PR soon. |
Also tested with AWS EC2 m5.8 instance, and found no error (master commit 3c404a5).
Env Configuration:
|
@pengzhao-intel I tried with branch v1.6.x @rongzha1 can you try with that branch too.
It seems this was run on an instance that supports AVX512. Can you try on an instance that has just AVX2 ? |
@pengzhao-intel |
You can try to use Also I just filed a PR related to slice op, but not sure if it will resolve this issue. Could you help double check? |
PR: #16737 |
Glad it works. |
Description
when MXNet is built for CPU MKL slice operator doesn't work.
Error Message
could not initialize a sub-memory
To Reproduce
Use MXNET cpu build with MKL and MKLDNN enabled from master
Steps to reproduce
(Paste the commands you ran that produced the error.)
MXNET_TEST_COUNT=1 nosetests --logging-level=DEBUG --verbose -s tests/nightly/test_large_array.py:test_slice
Environment
Ubuntu 16.04 DeepLearning AMI
We recommend using our script for collecting the diagnositc information. Run the following command and paste the outputs below:
The text was updated successfully, but these errors were encountered: