Skip to content

[PyTorch] fix unit8 in _convert_dtype_value#11924

Merged
masahi merged 1 commit intoapache:mainfrom
shingjan:torch_fix_unit8
Jun 27, 2022
Merged

[PyTorch] fix unit8 in _convert_dtype_value#11924
masahi merged 1 commit intoapache:mainfrom
shingjan:torch_fix_unit8

Conversation

@shingjan
Copy link

Hitting this error with vision_maskrcnn in torchbench. Stacktrace below:

ERROR:torchdynamo.optimizations.backends:tvm error
Traceback (most recent call last):
  File "/home/xx/torchdynamo/torchdynamo/optimizations/backends.py", line 640, in tvm_compile_inner
    mod, params = relay.frontend.from_pytorch(jit_mod, shape_list)
  File "/home/xx/tvm/python/tvm/relay/frontend/pytorch.py", line 4320, in from_pytorch
    outputs = converter.convert_operators(_get_operator_nodes(graph.nodes()), outputs, ret_name)
  File "/home/xx/tvm/python/tvm/relay/frontend/pytorch.py", line 3694, in convert_operators
    relay_out = relay_op(
  File "/home/xx/tvm/python/tvm/relay/frontend/pytorch.py", line 856, in new_full
    dtype = _convert_dtype_value(inputs[2])
  File "/home/xx/tvm/python/tvm/relay/frontend/pytorch.py", line 3765, in _convert_dtype_value
    return _convert_data_type(convert_torch_dtype_map[val])
  File "/home/xx/tvm/python/tvm/relay/frontend/pytorch.py", line 3806, in _convert_data_type
    raise NotImplementedError("input_type {} is not handled yet".format(input_type))
NotImplementedError: input_type torch.unit8 is not handled yet
input_type torch.unit8 is not handled yet

reference: https://pytorch.org/docs/stable/tensors.html

cc: @masahi

@masahi masahi merged commit e92db86 into apache:main Jun 27, 2022
@shingjan shingjan deleted the torch_fix_unit8 branch June 27, 2022 21:17
blackkker pushed a commit to blackkker/tvm that referenced this pull request Jul 7, 2022
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants