Commit 7c4fec605c488907ee3f88b68d15f3fa9a8d37d1

Authored by Josh Klontz
1 parent 32a4e672

fixed todo

Showing 1 changed file with 2 additions and 2 deletions
sdk/plugins/llvm.cpp
@@ -140,7 +140,7 @@ struct MatrixBuilder : public jit_matrix @@ -140,7 +140,7 @@ struct MatrixBuilder : public jit_matrix
140 } 140 }
141 141
142 std::vector<Value*> mallocArgs; 142 std::vector<Value*> mallocArgs;
143 - mallocArgs.push_back(elementsCode()); // TODO: FIX 143 + mallocArgs.push_back(bytesCode());
144 setData(b->CreateCall(malloc, mallocArgs)); 144 setData(b->CreateCall(malloc, mallocArgs));
145 } 145 }
146 146
@@ -165,7 +165,7 @@ struct MatrixBuilder : public jit_matrix @@ -165,7 +165,7 @@ struct MatrixBuilder : public jit_matrix
165 Value *singleFrameCode() const { return get(SingleFrame); } 165 Value *singleFrameCode() const { return get(SingleFrame); }
166 void setSingleFrameCode(bool singleFrame) const { setBit(singleFrame, SingleFrame); } 166 void setSingleFrameCode(bool singleFrame) const { setBit(singleFrame, SingleFrame); }
167 Value *elementsCode() const { return b->CreateMul(b->CreateMul(b->CreateMul(getChannels(), getColumns()), getRows()), getFrames()); } 167 Value *elementsCode() const { return b->CreateMul(b->CreateMul(b->CreateMul(getChannels(), getColumns()), getRows()), getFrames()); }
168 - Value *bytesCode() const { return b->CreateMul(b->CreateUDiv(bitsCode(), constant(8, 16)), elementsCode()); } 168 + Value *bytesCode() const { return b->CreateMul(b->CreateUDiv(b->CreateCast(Instruction::ZExt, bitsCode(), Type::getInt32Ty(getGlobalContext())), constant(8, 32)), elementsCode()); }
169 169
170 Value *columnStep() const { Value *columnStep = getChannels(); columnStep->setName(name+"_cStep"); return columnStep; } 170 Value *columnStep() const { Value *columnStep = getChannels(); columnStep->setName(name+"_cStep"); return columnStep; }
171 Value *rowStep() const { return b->CreateMul(getColumns(), columnStep(), name+"_rStep"); } 171 Value *rowStep() const { return b->CreateMul(getColumns(), columnStep(), name+"_rStep"); }