ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Codegen: isSupportedArgument check (#1417)

Browse Source 
* added isSupportedArgument check into codegen device op

* adding function call

* remove commented code
This commit is contained in:
arai713
2024-07-31 07:12:15 -07:00
committed by GitHub
parent b3f86e79dd
commit d32997a792
2 changed files with 181 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
codegen/src/device_grouped_conv_fwd_multiple_abd_operation_xdl_cshuffle.cpp
View File

@@ -233,6 +233,12 @@ extern "C" __global__ void run_${name}(
${BElementwiseOperation}{},
${CDEElementwiseOperation}{1.0f, 1.0f});
if(!DeviceConv::IsSupportedArgument(arg))
{
printf("Arguement is not supported.\n");
return;
};
constexpr ck::LoopScheduler LoopSched = ck::make_default_loop_scheduler();
// GridwiseGemm