Commit 3ce8237d by liyinqiao Committed by 李垠桥

Bug fixed.

Boundary detection (order >= 0) bug fixed.
parent 6803ce02
......@@ -874,7 +874,6 @@ XTensor * NewTensor(const XTensor * a, bool isFilledData)
a->devID, a->mem);
return newTensor;
}
/*
......
......@@ -233,7 +233,7 @@ void _MultiplyBroadcast(const XTensor * a, const XTensor * b, XTensor * c, DTYPE
{
CheckNTErrors(a->order == b->order, "Wrong tensor orders!");
CheckNTErrors(a->order == c->order, "Wrong tensor orders!");
CheckNTErrors(a->order > 0, "TODO!");
CheckNTErrors(a->order >= 0, "TODO!");
int order = a->order;
int count = 0;
......
......@@ -220,7 +220,7 @@ void _SumBroadcast(const XTensor * a, const XTensor * b, XTensor * c, DTYPE beta
{
CheckNTErrors(a->order == b->order, "Wrong tensor orders!");
CheckNTErrors(a->order == c->order, "Wrong tensor orders!");
CheckNTErrors(a->order > 0, "TODO!");
CheckNTErrors(a->order >= 0, "TODO!");
int order = a->order;
int count = 0;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论