Created
March 26, 2025 11:01
-
-
Save shauray8/34f4d9248350e9ab804a1f08cba47c2a to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Graph break in user code at /workspace/Wan2GP/wan/modules/sage2_core.py:66 | |
| Reason: Unsupported: torch.* op returned non-Tensor int call_function <function device_count at 0x795533c18400> | |
| User code traceback: | |
| File "/workspace/Wan2GP/wan/modules/model.py", line 452, in forward | |
| y = self.self_attn( xlist, seq_lens, grid_sizes,freqs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/nn/modules/module.py", line 1750, in _call_impl | |
| return forward_call(*args, **kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/mmgp/offload.py", line 1926, in check_empty_cuda_cache | |
| return previous_method(*args, **kwargs) | |
| File "/workspace/Wan2GP/wan/modules/model.py", line 266, in forward | |
| x = pay_attention( | |
| File "/workspace/Wan2GP/wan/modules/attention.py", line 245, in pay_attention | |
| x = sageattn_wrapper(qkv_list, lq).unsqueeze(0) | |
| File "/workspace/Wan2GP/wan/modules/attention.py", line 61, in sageattn_wrapper | |
| o = alt_sageattn(qkv_list, tensor_layout="NHD").squeeze(0) | |
| File "/workspace/Wan2GP/wan/modules/sage2_core.py", line 133, in sageattn | |
| arch = get_cuda_arch_versions()[qkv_list[0].device.index] | |
| File "/workspace/Wan2GP/wan/modules/sage2_core.py", line 66, in get_cuda_arch_versions | |
| for i in range(torch.cuda.device_count()): | |
| Traceback (most recent call last): | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/lazy.py", line 170, in realize_and_forward | |
| return getattr(self.realize(), name)(*args, **kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/nn_module.py", line 914, in call_function | |
| return variables.UserFunctionVariable(fn, source=source).call_function( | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1736, in CALL_FUNCTION_EX | |
| self.call_function(fn, argsvars.items, kwargsvars) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/lazy.py", line 170, in realize_and_forward | |
| return getattr(self.realize(), name)(*args, **kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 858, in call_function | |
| return self.func.call_function(tx, merged_args, merged_kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1736, in CALL_FUNCTION_EX | |
| self.call_function(fn, argsvars.items, kwargsvars) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 378, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 317, in call_function | |
| return super().call_function(tx, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/functions.py", line 118, in call_function | |
| return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 903, in inline_user_function_return | |
| return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3072, in inline_call | |
| return cls.inline_call_(parent, func, args, kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 3198, in inline_call_ | |
| tracer.run() | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 1052, in run | |
| while self.step(): | |
| ^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 962, in step | |
| self.dispatch_table[inst.opcode](self, inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 659, in wrapper | |
| return inner_fn(self, inst) | |
| ^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2341, in CALL | |
| self._call(inst) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 2335, in _call | |
| self.call_function(fn, args, kwargs) | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/symbolic_convert.py", line 897, in call_function | |
| self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/torch.py", line 953, in call_function | |
| tensor_variable = wrap_fx_proxy( | |
| ^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/builder.py", line 2153, in wrap_fx_proxy | |
| return wrap_fx_proxy_cls(target_cls=TensorVariable, **kwargs) | |
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/builder.py", line 2219, in wrap_fx_proxy_cls | |
| return _wrap_fx_proxy( | |
| ^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/builder.py", line 2317, in _wrap_fx_proxy | |
| return handle_traced_output( | |
| ^^^^^^^^^^^^^^^^^^^^^ | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/variables/builder.py", line 2517, in handle_traced_output | |
| unimplemented( | |
| File "/usr/local/lib/python3.11/dist-packages/torch/_dynamo/exc.py", line 317, in unimplemented | |
| raise Unsupported(msg, case_name=case_name) | |
| torch._dynamo.exc.Unsupported: torch.* op returned non-Tensor int call_function <function device_count at 0x795533c18400> |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment