x10000year opened a new issue #7934: Bug of group2ctx? Wrong device placement? URL: https://github.com/apache/incubator-mxnet/issues/7934 For the following code: x = mx.symbol.MyOp() exe = x.bind(mx.gpu(), {}, group2ctx={"a": mx.cpu(), "b": mx.gpu()}) exe.forward() where MyOp is a custom operator written in c++, which prints "CPU" if it is run in cpu context, or prints "GPU" if run in gpu context. I don't use mx.AttrScope to specify the group of x, so default context should be used for x. However, the above code prints "CPU", which means that x is run in cpu context. Why? If I set group2ctx={"b": mx.cpu(), "a": mx.gpu()}. Then it prints "GPU". Basically, I found that the group that has the alphabetically smaller name is chosen for x. Very strange. Is this a bug? ---------------------------------------------------------------- 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: us...@infra.apache.org
With regards, Apache Git Services
