added support for fp16 constant matricies

biometrics · Jul 3, 2016 · 08dc7ad · 08dc7ad
1 parent 338e614
commit 08dc7ad
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/backend.cpp b/src/backend.cpp
@@ -1245,8 +1245,9 @@ struct ConstantMat : public likely_expression
                                       : (llvmType == likely_u16) ? ConstantDataArray::get(builder.module->context->context, ArrayRef<uint16_t>((uint16_t*) &m->data, elements))
                                       : (llvmType == likely_u32) ? ConstantDataArray::get(builder.module->context->context, ArrayRef<uint32_t>((uint32_t*) &m->data, elements))
                                       : (llvmType == likely_u64) ? ConstantDataArray::get(builder.module->context->context, ArrayRef<uint64_t>((uint64_t*) &m->data, elements))
-                                      : (llvmType == likely_f32) ? ConstantDataArray::get(builder.module->context->context, ArrayRef<float   >((float*)    &m->data, elements))
-                                      :                            ConstantDataArray::get(builder.module->context->context, ArrayRef<double  >((double*)   &m->data, elements)) };
+                                      : (llvmType == likely_f16) ? ConstantDataArray::getFP(builder.module->context->context, ArrayRef<uint16_t>((uint16_t*) &m->data, elements))
+                                      : (llvmType == likely_f32) ? ConstantDataArray::getFP(builder.module->context->context, ArrayRef<uint32_t>((uint32_t*) &m->data, elements))
+                                      :                            ConstantDataArray::getFP(builder.module->context->context, ArrayRef<uint64_t>((uint64_t*) &m->data, elements)) };
         Constant *const constantStruct = ConstantStruct::get(structType, values);
 
         stringstream name;