Traceback (most recent call last):


 "<stdin>", line 1, in <module>


 "/Users/wangy1/Documents/GitHub/jax/jax/_src/dlpack.py", line 42, in to_dlpack

buf, take_ownership=take_ownership)
RuntimeError: Unimplemented: XLA type PRED has no DLPack equivalent

Traceback (most recent call last):


 "experiments/segmentation/demo.py", line 16, in <module>

output = model.evaluate(img)


 "/cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/models/base.py", line 78, in evaluate

pred = self.forward(x)


 "/cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/models/fcn.py", line 51, in forward

_, _, c3, c4 = self.base_forward(x)


 "/cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/models/base.py", line 67, in base_forward

x = self.pretrained.conv1(x)


 "/home/nakka/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__

result = self.forward(*input, **kwargs)


 "/home/nakka/.local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward

input = module(input)


 "/home/nakka/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__

result = self.forward(*input, **kwargs)


 "/cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/nn/syncbn.py", line 122, in forward

self.activation, self.slope).view(input_shape)


 "/cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/functions/syncbn.py", line 95, in forward

y = lib.gpu.batchnorm_forward(x, _ex, _exs, gamma, beta, ctx.eps)
RuntimeError: cudaGetLastError() == cudaSuccess ASSERT FAILED at /cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/lib/gpu/syncbn_kernel.cu:289, please report a bug to PyTorch. (BatchNorm_Forward_CUDA at /cvlabdata2/home/krishna/packages/conda2/envs/py3.6/lib/python3.6/site-packages/encoding/lib/gpu/syncbn_kernel.cu:289)
