Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

[Bug] relax.build crashes with an AttributeError during the dlight GPU scheduling pass #20047

Copy link
Copy link

Description

@XYZboom
Issue body actions

Expected behavior

relax.build(mod, target="cuda") should compile successfully for any valid Relax IR module containing relax.op.nn.conv2d. The compiled module should execute and produce correct convolution results.

Actual behavior

relax.build crashes with an AttributeError during the dlight GPU scheduling pass:

AttributeError: 'FloorDiv' object has no attribute 'value'

Environment

  • OS: Linux (x86_64, conda environment)
  • GPU: NVIDIA GeForce RTX 3080 Ti (12GB VRAM)
  • CUDA driver: 580.76.05
  • TVM version: 0.25.0.post1 (installed via pip)
  • Target: cuda (GPU compilation)
  • Python: 3.12

Steps to reproduce

from tvm import relax

bb = relax.BlockBuilder()
v = relax.Var("v", relax.TensorStructInfo(
    shape=relax.ShapeExpr([1, 1, 3, 10]), dtype="float32"))
w = relax.Var("w", relax.TensorStructInfo(
    shape=relax.ShapeExpr([1, 1, 1, 2]), dtype="float32"))
with bb.function("f", [v, w]):
    c = bb.emit(relax.op.nn.conv2d(v, w,
        strides=[1, 1], padding=[0, 0], dilation=[1, 1], groups=1))
    bb.emit_func_output(c)
mod = bb.get()
ex = relax.build(mod, target="cuda")

Triage

  • bug
  • CUDA
  • dlight
  • gemv
  • conv2d
  • needs-triage
  • needs-triage
Reactions are currently unavailable

Metadata

Metadata

Assignees

No one assigned

    Labels

    needs-triagePRs or issues that need to be investigated by maintainers to find the right assignees to address itPRs or issues that need to be investigated by maintainers to find the right assignees to address ittype: bug

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions

    Morty Proxy This is a proxified and sanitized view of the page, visit original site.