This commit is contained in:
Michal Moskal 2020-10-13 13:59:02 +02:00
Родитель b45baa020c
Коммит 83641064fd
2 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -17,6 +17,7 @@ interface CmdOptions {
sampleModel?: string
testAll?: boolean
optimize?: boolean
float16?: boolean
}
let options: CmdOptions
@ -26,6 +27,7 @@ function getCompileOptions(): Options {
optimize: options.optimize,
verbose: options.debug,
includeTest: options.testData,
float16weights: options.float16,
}
}
@ -177,6 +179,7 @@ export async function mainCli() {
.option("-d, --debug", "enable debugging")
.option("-n, --no-validate", "don't validate resulting model")
.option("-g, --no-optimize", "don't optimize IR")
.option("-h, --float16", "use float16 weights")
.option("-t, --test-data", "include test data in binary model")
.option("-T, --test-all", "test all included sample models")
.option("-s, --sample-model <name>", "use an included sample model")

Просмотреть файл

@ -130,6 +130,7 @@ function addFloat16(mi: ModelInfo, v: number) {
export function alignWeights(mi: ModelInfo) {
while (mi.weightPtr & 3)
addParamBytes(mi, [0])
mi.weightAsm += ".align 4\n"
}
export function addWeight(mi: ModelInfo, v: number) {