The GitHub Actions job "CI" on 
tvm.git/refactor-and-introduce-allocbuffer-and-phase-out-allocatenode has 
failed.
Run started by GitHub user tqchen (triggered by tqchen).

Head commit for run:
fe7c648bc92607be6b08c82c2ef3e3e28877d9e6 / tqchen <[email protected]>
[TIR] Add VisitBufferDef/VisitBufferUse to base StmtVisitor/StmtMutator

Introduce VisitBufferDef and VisitBufferUse virtual methods in the base
StmtVisitor and StmtMutator classes to properly visit buffer fields
(shape, strides, elem_offset) at definition and use sites.

Previously, passes that do variable substitution (e.g., IRSubstitute)
would miss buffer fields, causing orphan variables after substitution.
This fixes issues like test_dispatch_cumsum_gpu failing with
"variables [m, n] are used but not passed in as API arguments".

Key design:
- VisitBufferDef: called at definition sites (AllocBuffer, DeclBuffer,
  SBlock alloc_buffers/match_buffers). Visits shape/strides/elem_offset
  via VisitExpr. Does NOT visit data (it is a Var definition).
- VisitBufferUse: called at use sites (BufferStore, BufferLoad, SBlock
  reads/writes). Base just checks buffer_remap_ for remapped buffers.
- buffer_remap_ (ffi::Map<Buffer, Buffer>): populated by VisitBufferDef
  when buffer fields change.

Subclass updates:
- VarUseDefAnalyzer: piggybacks on VisitBufferDef/VisitBufferUse for
  HandleDef/HandleUse instead of separate node overrides.
- IRSubstitute: VisitBufferDef additionally handles data var; removed
  RemapBuffer helper. VisitBufferUse delegates to base buffer_remap_.
- IndexDataTypeRewriter: VisitBufferDef sets is_enabled_ flag then
  delegates to base; removed VisitBuffer/GetRemappedBuffer/buffer_remap_.
- Simplifier: removed CollectVarsUsedInBufferDefinition workaround since
  buffer fields are now properly updated via VisitBufferDef. Overrides
  VisitBufferDef to also substitute data vars.
- PrimFuncSpecializer: overrides VisitBufferUse for its own buffer_map_.
- RenewDefs: overrides VisitBufferDef/VisitBufferUse for def/use logic.

Report URL: https://github.com/apache/tvm/actions/runs/22633785870

With regards,
GitHub Actions via GitBox


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to