Bf16 Verify Onnx Tests

src/onnx/onnx_parser.cpp

@@ -687,7 +687,7 @@ shape::type_t get_type(int dtype)
     case 22: return shape::int8_type;
     case 14:
     case 15:
-    case 16:
+    case 16: return shape::bf16_type;
     case 17:
     case 19:
     case 20:
@@ -700,7 +700,8 @@ shape::type_t get_type(int dtype)
 bool is_type_float(shape::type_t dtype)
 {
     bool r = false;
-    if(dtype == shape::float_type or dtype == shape::double_type or dtype == shape::half_type)
+    if(dtype == shape::float_type or dtype == shape::double_type or dtype == shape::half_type or
+       dtype == shape::bf16_type)
     {
         r = true;
     }

src/onnx/parse_instancenorm.cpp

@@ -36,7 +36,8 @@ namespace onnx {
 
 struct parse_instancenorm : op_parser<parse_instancenorm>
 {
-    std::set<shape::type_t> valid_types = {shape::float_type, shape::half_type, shape::double_type};
+    std::set<shape::type_t> valid_types = {
+        shape::float_type, shape::half_type, shape::double_type, shape::bf16_type};
 
     std::vector<op_desc> operators() const { return {{"InstanceNormalization"}}; }
 

test/onnx/add_bf16_test.onnx

@@ -0,0 +1,16 @@
+	add_bf16_test:R
+
+
0
+
1
2"Addadd_bf16_testZ
+
0
+
+
+
Z
+
1
+
+
+
b
+
2
+
+
+
B

test/onnx/eyelike_bf16_test.onnx

@@ -0,0 +1,11 @@
+	eyelike_bf16_test:R
+
+T1T2"EyeLikeeyelike_bf16_testZ
+T1
+
+
+b
+T2
+
+
+B

test/onnx/gelu_default_bf16_test.onnx

@@ -0,0 +1,11 @@
+	gelu_default_bf16_test:P
+
+
x
y"Gelugelu_default_bf16_testZ
+
x
+
+
+b
+
y
+
+
+B