composable_kernel
composable_kernel copied to clipboard
[Issue]: in gemm_basic example, throw std::runtime_error() hangs with non-supported arg shape
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