composable_kernel icon indicating copy to clipboard operation
composable_kernel copied to clipboard

[Issue]: in gemm_basic example, throw std::runtime_error() hangs with non-supported arg shape

Open ZJLi2013 opened this issue 9 months ago • 0 comments

Problem Description

in example/ck_tile/03_gemm/gemm_basic.cpp

with not supported args, e..g -m=128, -n=128, -k=32, it will hangs at :

    if(!Kernel::IsSupportedArgument(kargs))
    {
        throw std::runtime_error("Wrong! Arguments not supported! Skipping gemm!\n");
    }

basically it can print out "Wrong ! Argument not supported", but hangs then, rather than exit as expected, make the user need to restart the container ..

the following can fix the hang issue for now

    if(!Kernel::IsSupportedArgument(kargs))
    {
        // throw std::runtime_error("Wrong! Arguments not supported! Skipping gemm!\n");
        std::cerr << "Wrong! Arguments not supported! Skipping gemm!" << std::endl;
        std::exit(EXIT_FAILURE);
    }

Operating System

Ubuntu 22.04

CPU

AMD

GPU

AMD Instinct MI300

Other

No response

ROCm Version

ROCm 6.0.0

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response

ZJLi2013 avatar Mar 03 '25 08:03 ZJLi2013