Skip to content

Commit

Permalink
x64: pool: Accuracy issue in benchnn pooling test
Browse files Browse the repository at this point in the history
  • Loading branch information
jstachowintel committed Feb 12, 2025
1 parent 6c5e592 commit f81b3a7
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/cpu/x64/jit_uni_pooling.hpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*******************************************************************************
* Copyright 2017-2024 Intel Corporation
* Copyright 2017-2025 Intel Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -75,6 +75,11 @@ struct jit_uni_pooling_fwd_t : public primitive_t {
CHECK(jit_uni_pool_kernel<isa>::init_conf(
jpp_, scratchpad, attr_, this));

dim_t src_size = static_cast<dim_t>(jpp_.mb) * jpp_.c * jpp_.id * jpp_.ih * jpp_.iw;

VDISPATCH_POOLING_IC(src_size <= INT_MAX, VERBOSE_UNSUPPORTED_FEATURE,
"src size > INT_MAX is not supported");

return status::success;
}

Expand Down

0 comments on commit f81b3a7

Please sign in to comment.