[GPU] Skip kernel logic for Concat fix (#16885)
This commit is contained in:
1 parent
da7ee613a3
commit
656428bc4f
4 files changed
+3
-10
No files matched your search
+2
-6
@@ -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});
|
||||
}
|
||||
|
||||
-1
@@ -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
|
||||
+1
-1
@@ -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;
|
||||
|
||||
Reference in new issue
Block a user