Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

compilation failure for simple f32 matmul using the cuda backend #19575

Open
ziereis opened this issue Dec 31, 2024 · 0 comments
Open

compilation failure for simple f32 matmul using the cuda backend #19575

ziereis opened this issue Dec 31, 2024 · 0 comments
Labels
bug 🐞 Something isn't working codegen/nvvm NVVM code generation compiler backend

Comments

@ziereis
Copy link
Contributor

ziereis commented Dec 31, 2024

What happened?

compiling this example:

func.func @matmul(%arg0: tensor<4096x4096xf32>, %arg1: tensor<4096x4096xf32>) -> tensor<4096x4096xf32> {
    %0 = tensor.empty() : tensor<4096x4096xf32>
    %cst = arith.constant 0.000000e+00 : f32
    %1 = linalg.fill ins(%cst : f32) outs(%0 : tensor<4096x4096xf32>) -> tensor<4096x4096xf32>
    %2 = linalg.matmul ins(%arg0, %arg1 : tensor<4096x4096xf32>, tensor<4096x4096xf32>) outs(%1 : tensor<4096x4096xf32>) -> tensor<4096x4096xf32>
    return %2 : tensor<4096x4096xf32>
}

fails with:

example.mlir:5:10: error: 'func.func' op uses 229376 bytes of shared memory; exceeded the limit of 166912 bytes

Steps to reproduce your issue

iree-compile --iree-hal-target-device=cuda --iree-cuda-target=sm_86

What component(s) does this issue relate to?

Compiler

Version information

f27feff

Additional context

No response

@ziereis ziereis added the bug 🐞 Something isn't working label Dec 31, 2024
@ScottTodd ScottTodd added the codegen/nvvm NVVM code generation compiler backend label Jan 2, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug 🐞 Something isn't working codegen/nvvm NVVM code generation compiler backend
Projects
None yet
Development

No branches or pull requests

2 participants