diff options
Diffstat (limited to 'src/gpu/cl/operators/ClMatMul.h')
-rw-r--r-- | src/gpu/cl/operators/ClMatMul.h | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/src/gpu/cl/operators/ClMatMul.h b/src/gpu/cl/operators/ClMatMul.h index 64dcf217bd..1733def21c 100644 --- a/src/gpu/cl/operators/ClMatMul.h +++ b/src/gpu/cl/operators/ClMatMul.h @@ -21,15 +21,14 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef ACL_SRC_GPU_CL_OPERATORS_CLMATMUL -#define ACL_SRC_GPU_CL_OPERATORS_CLMATMUL +#ifndef ACL_SRC_GPU_CL_OPERATORS_CLMATMUL_H +#define ACL_SRC_GPU_CL_OPERATORS_CLMATMUL_H #include "arm_compute/function_info/ActivationLayerInfo.h" #include "arm_compute/function_info/MatMulInfo.h" +#include "src/gpu/cl/IClKernel.h" #include "src/gpu/cl/IClOperator.h" -#include "src/gpu/cl/kernels/ClMatMulLowpNativeKernel.h" -#include "src/gpu/cl/kernels/ClMatMulNativeKernel.h" #include <memory> @@ -95,11 +94,8 @@ public: void run(ITensorPack &tensors) override; private: - std::unique_ptr<kernels::ClMatMulNativeKernel> _matmul_native_kernel{nullptr}; - std::unique_ptr<kernels::ClMatMulLowpNativeKernel> _matmul_lowp_native_kernel{nullptr}; - - bool _is_quantized{false}; + std::unique_ptr<opencl::IClKernel> _matmul_kernel{nullptr}; }; } // namespace opencl } // namespace arm_compute -#endif /* ACL_SRC_GPU_CL_OPERATORS_CLMATMUL */ +#endif // ACL_SRC_GPU_CL_OPERATORS_CLMATMUL_H |