gpu: jit: enable dnnl::impl::bfloat16_t in the generator

src/gpu/jit/jit_generator.hpp

@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright 2019-2021 Intel Corporation
+* Copyright 2019-2022 Intel Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -19,15 +19,18 @@
 
 #include <memory>
 
+#include "common/bfloat16.hpp"
 #include "common/float16.hpp"
 #include "common/nstl.hpp"
 #include "gpu/jit/jit_generator_base.hpp"
 #include "oneapi/dnnl/dnnl_config.h"
 
 namespace ngen {
+using bfloat16 = dnnl::impl::bfloat16_t;
 using half = dnnl::impl::float16_t;
 }
 
+#define NGEN_BFLOAT16_TYPE
 #define NGEN_HALF_TYPE
 
 #include "gpu/jit/ngen/ngen_opencl.hpp"