Lunderberg opened a new pull request, #14495: URL: https://github.com/apache/tvm/pull/14495
Previously, `kDeviceThreadAxis` defined the IterVar to be used for each thread/block axis, and `kUseDynamicSharedMemoryTag` defined whether dynamic memory allocations exist, which are primarily used to produce a list of strings by `tvm::codegen::ExtractFuncInfo`. Because `kDeviceThreadAxis` is a `Array<IterVar>`, the IterVar is used prior to its definition site at `tir::attr::thread_extent`, which results in errors when attempting to round-trip through TVMScript. This commit replaces these attributes with `attr::kKernelLaunchParams`, which directly contains the kernel launch parameters. These are expressed as an `Array<String>`, allowing the generated TVMScript to successfully round-trip. This is part of changes described in https://github.com/apache/tvm/pull/14486, to improve round-trip TVMScript failures that occur in lowering. -- 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]
