-
Notifications
You must be signed in to change notification settings - Fork 68
[Bug] operator.gt got an object with type <class 'int'> #320
Copy link
Copy link
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
Hi @yaoyaoding, I encounter a bug when I run T5Model with hidet:
DEBUG:hidet.graph.frontend.torch.interpreter:interpreting node 34: %gt : [#users=1] = call_function[target=operator.gt](args = (%sub_1, 0), kwargs = {})
Traceback (most recent call last):
File "hidet-path/python/hidet/graph/frontend/torch/interpreter.py", line 260, in forward
hidet_env[node.name] = hidet_func(*hidet_args, **hidet_kwargs)
File "hidet-path/python/hidet/graph/frontend/torch/register_functions.py", line 694, in gt
return ops.greater(a, b)
File "hidet-path/python/hidet/graph/ops/definitions/compare.py", line 85, in greater
return GreaterOp(x, y).get_output(0)
File "hidet-path/python/hidet/graph/ops/definitions/compare.py", line 34, in __init__
super().__init__(x, y, lambda a, b: a > b, name='gt')
File "hidet-path/python/hidet/graph/ops/definitions/arithmetic.py", line 125, in __init__
task=BinaryElementwiseTask(name, input_like(x, 'x'), input_like(y, 'y'), op=op),
File "hidet-path/python/hidet/graph/ops/definitions/utils/tensor_utils.py", line 26, in input_like
raise TypeError('Expect a hidet.Tensor, but got an object with type {}'.format(type(tensor)))
TypeError: Expect a hidet.Tensor, but got an object with type <class 'int'>
Could you please help me fix it?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working