[GPU] Skip kernel logic for Concat fix (#16885)

This commit is contained in:
Roman Lyamin authored and GitHub committed 2023-04-13 15:55:05 +04:00
1 parent da7ee613a3
commit 656428bc4f
4 files changed
+3 -10

No files matched your search

@@ -109,10 +109,6 @@ ConcatenationKernelBase::DispatchData ConcatenationKernelBase::SetDefault(const
return dispatchData;
}
bool ConcatenationKernelBase::SkipKernelExecution(const concatenation_params& params, size_t kernel_id) const {
return params.inputs[kernel_id].LogicalSize() == 0;
}
KernelsData ConcatenationKernelBase::GetCommonKernelsData(const Params& params, const optional_params& options) const {
if (!Validate(params, options)) {
return {};
@@ -140,7 +136,7 @@ KernelsData ConcatenationKernelBase::GetCommonKernelsData(const Params& params,
DispatchData dispatchData = SetDefault(newParams);
kernel.params.workGroups.global = dispatchData.gws;
kernel.params.workGroups.local = dispatchData.lws;
kernel.skip_execution = SkipKernelExecution(prim_params, i);
kernel.skip_execution = KernelData::SkipKernelExecution(newParams);
ScalarDescriptor s;
s.t = ScalarDescriptor::Types::UINT32;
@@ -178,7 +174,7 @@ KernelsData ConcatenationKernelBase::GetCommonKernelsData(const Params& params,
kernel.code.kernelString = GetKernelString(kernelName, jit, entryPoint, params.engineInfo);
kernel.params.workGroups.global = dispatchData.gws;
kernel.params.workGroups.local = dispatchData.lws;
kernel.skip_execution = SkipKernelExecution(newParams, i);
kernel.skip_execution = KernelData::SkipKernelExecution(newParams);
if (is_dynamic) {
kernel.params.arguments.push_back({ArgumentDescriptor::Types::SHAPE_INFO, 0});
}
@@ -68,6 +68,5 @@ protected:
virtual size_t GetAlignment(const concatenation_params& /*params*/) const {
return 1;
}
bool SkipKernelExecution(const concatenation_params& params, size_t kernel_id) const;
};
} // namespace kernel_selector
@@ -114,7 +114,7 @@ KernelsData ConcatenationKernel_fs_b_yx_fsv32::GetKernelsData(const Params& para
kernel.params.workGroups.local = dispatchData.lws;
kernel.params.arguments.push_back({ArgumentDescriptor::Types::INPUT, (uint32_t)i});
kernel.params.arguments.push_back({ArgumentDescriptor::Types::OUTPUT, 0});
kernel.skip_execution = SkipKernelExecution(newParams, i);
kernel.skip_execution = KernelData::SkipKernelExecution(newParams);
ScalarDescriptor s;
s.t = ScalarDescriptor::Types::UINT32;
@@ -459,7 +459,6 @@ template<typename T>
void generic_lstm_gpu_test(int layers, int sequence_len, int direction, int batch_size, int input_size, int hidden_size,
bool hasBias, bool hasInitialHidden, bool hasInitialCell,
T clip_threshold, bool input_forget, bool is_caching_test = false) {
GTEST_SKIP(); // Issue: 107984
std::cout << "Layers = " << layers << " Input Size = " << input_size << " Hidden Size = " << hidden_size
<< " Sequence Len = " << sequence_len << " Direction = " << direction << " Batch Size = " << batch_size << std::endl;
int min_random = -2, max_random = 2;
@@ -639,7 +638,6 @@ void generic_lstm_gpu_test(int layers, int sequence_len, int direction, int batc
// -------------------------------------------------------
template<typename T>
void lstm_gpu_output_test(const lstm_output_selection& output_selection, int directions, bool is_caching_test = false) {
GTEST_SKIP(); // Issue: 107984
int layers = 1;
int sequence_len = 4;
int batch_size = 3;