Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Bug] 使用自定义的算子的时候报错:[ERROR](GRAPH:OneflowGraph_0:OneflowGraph) building graph got error. #1161

Open
woshiame opened this issue Dec 11, 2024 · 0 comments
Labels
Request-bug Something isn't working

Comments

@woshiame
Copy link

woshiame commented Dec 11, 2024

Your current environment information

cuda 12.1
onediff 1.2.0.dev202407230130
onediffx 1.1.0
oneflow 0.9.1.dev20241017+cu121
onefx 0.0.3

🐛 Describe the bug

我的代码部分如下:
def _check_legality(self, input: Tensor) -> bool:
print("当前 torch 版本号:", torch.version)
is_nhwc = input.is_contiguous(memory_format=torch.channels_last)
输出显示torch的版本号似乎最后被改为和oneflow一致,以下为输出:
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
当前 torch 版本号: 2.3.0+cu121
0%| | 0/100 [00:00<?, ?it/s]当前 torch 版本号: 0.9.1.dev20241017+cu121
ERROR building graph got error.
ERROR [2024-12-11 14:11:31] /usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/deployable_module.py:44 - Exception in wrapper: e=TypeError('is_contiguous() takes no keyword arguments')
WARNING [2024-12-11 14:11:31] /usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/deployable_module.py:45 - Recompile oneflow module ...
当前 torch 版本号: 0.9.1.dev20241017+cu121
ERROR building graph got error.

@woshiame woshiame added the Request-bug Something isn't working label Dec 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Request-bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant