Closed Lunderberg closed 2 months ago
@tvm-bot rerun
Failed to re-run CI in https://github.com/apache/tvm/actions/runs/9245501195
URI: | /job/tvm-arm/job/PR-17024/buildWithParameters |
---|---|
STATUS: | 404 |
MESSAGE: | Not Found |
SERVLET: | Stapler |
due to recent ci restart, likely we need to close and re create another pr for the ci to get retriggerd
Thank you for looking into the CI failures, and I'll resubmit as a new PR. Looks like #17022 and #17023 have the similar CI failure, as they were all submitted while the CI was down.
The
ComputePrimValue
transform is used to compute the value of symbolic expressions that may appear within a Relax function. For example, to compute a boolean condition used for arelax::If
node. These functions are used for small host-side computations, prior to launching a device kernel.This commit updates
ComputePrimValue
to annotate the generatedPrimFunc
withtir::attr::kIsHostFunc
. This annotation is required for correct behavior intvm.dlight.ApplyDefaultSchedule
, to avoid erroneous scheduling of this function for the GPU, and fortir::transform::BindTarget
, to ensure that the function is compiled for execution on the host.