Hi,
I’m trying to run nodeflow exampale (1_sampling_mx.py) with torch’s tensors instead of mx and mx.block. I keep getting “raise DGLError(py_str(_LIB.DGLGetLastError())) dgl._ffi.base.DGLError: [20:28:29] /Users/xiangsx/work/dgl/dgl/src/kernel/cpu/…/binary_reduce_impl.h:112: Unsupported dtype: int64”
In the error stack, the following line seems to be the offender.
" g.update_all(message_func=fn.copy_src(src=‘h’, out=‘m’),
reduce_func=fn.sum(msg=‘m’, out=‘h’),
apply_node_func=lambda node: {‘h’: layersi[‘activation’]})"
By the way, I had to comment out layer.initialize() because torch.nn.module doesn’t have it as mx.Block has.
Any idea how I could be fixing it?
Thanks so much.