AD-lite24 commented on issue #17577:
URL: https://github.com/apache/tvm/issues/17577#issuecomment-2567146397
` for (auto [gvar, base_func] : mod->functions) {
ICHECK(base_func->IsInstance<PrimFuncNode>()) << "CodeGenOpenCL: Can
only take PrimFunc";
auto prim_func = Downcast<PrimFunc>(base_func);
auto calling_conv = prim_func->GetAttr<Integer>(tvm::attr::kCallingConv);
ICHECK(calling_conv == CallingConv::kDeviceKernelLaunch)
<< "CodeGenOpenCL: expect calling_conv equals
CallingConv::kDeviceKernelLaunch";
functions.Set(gvar, prim_func);
}`
In the pytorch conversion to IRModule, apparently the calling_conv attr was
never set to kDeviceKernelLaunch
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
