Kaydet (Commit) ec4b9fb2 authored tarafından fengzeng's avatar fengzeng Kaydeden (comit) I-Jui (Ray) Sung

GPU Calc: Optimized PMT

AMLOEXT-266

Change-Id: Ia55a379aacb820b088ffd7e22409974918f9193a
Signed-off-by: 's avatarhaochen <haochen@multicorewareinc.com>
Signed-off-by: 's avatarWei Wei <weiwei@multicorewareinc.com>
Signed-off-by: 's avatarI-Jui (Ray) Sung <ray@multicorewareinc.com>
üst 9a4bc254
...@@ -2451,110 +2451,72 @@ void OpPMT::GenSlidingWindowFunction(std::stringstream &ss, ...@@ -2451,110 +2451,72 @@ void OpPMT::GenSlidingWindowFunction(std::stringstream &ss,
ss << ", "; ss << ", ";
vSubArguments[i]->GenSlidingWindowDecl(ss); vSubArguments[i]->GenSlidingWindowDecl(ss);
} }
ss << ") {\n "; ss<<") {\n";
ss << "double tmp = 0;\n "; ss<<" double tmp = 0;\n";
ss << "int gid0 = get_global_id(0);\n "; ss<<" double temp=0.0;\n";
ss<<"double tmp0,tmp1,tmp2;\n "; ss<<" int gid0 = get_global_id(0);\n";
ss<<"double tmp3=0,tmp4=0;\n "; ss<<" double tmp0=0,tmp1=0,tmp2=0;\n";
ss<<" double tmp3=0,tmp4=0;\n";
size_t i = vSubArguments.size(); size_t i = vSubArguments.size();
size_t nItems = 0;
ss <<"\n "; ss <<"\n ";
//while (i-- > 1) //while (i-- > 1)
for (i = 0; i < vSubArguments.size(); i++) for (i = 0; i < vSubArguments.size(); i++)
{ {
FormulaToken *pCur = vSubArguments[i]->GetFormulaToken(); FormulaToken *pCur = vSubArguments[i]->GetFormulaToken();
assert(pCur); assert(pCur);
if (pCur->GetType() == formula::svDoubleVectorRef) if (pCur->GetType() == formula::svSingleVectorRef)
{
const formula::DoubleVectorRefToken* pDVR =
dynamic_cast<const formula::DoubleVectorRefToken *>(pCur);
size_t nCurWindowSize = pDVR->GetRefRowSize();
ss << "for (int i = ";
if (!pDVR->IsStartFixed() && pDVR->IsEndFixed()) {
#ifdef ISNAN
ss << "gid0; i < " << pDVR->GetArrayLength();
ss << " && i < " << nCurWindowSize << "; i++){\n ";
#else
ss << "gid0; i < "<< nCurWindowSize << "; i++)\n ";
#endif
} else if (pDVR->IsStartFixed() && !pDVR->IsEndFixed()) {
#ifdef ISNAN
ss << "0; i < " << pDVR->GetArrayLength();
ss << " && i < gid0+"<< nCurWindowSize << "; i++){\n ";
#else
ss << "0; i < gid0+"<< nCurWindowSize << "; i++)\n ";
#endif
} else if (!pDVR->IsStartFixed() && !pDVR->IsEndFixed()){
#ifdef ISNAN
ss << "0; i + gid0 < " << pDVR->GetArrayLength();
ss << " && i < "<< nCurWindowSize << "; i++){\n ";
#else
ss << "0; i < "<< nCurWindowSize << "; i++)\n ";
#endif
}
else {
#ifdef ISNAN
ss << "0; i < "<< nCurWindowSize << "; i++){\n ";
#else
ss << "0; i < "<< nCurWindowSize << "; i++)\n ";
#endif
}
nItems += nCurWindowSize;
}
else if (pCur->GetType() == formula::svSingleVectorRef)
{ {
#ifdef ISNAN #ifdef ISNAN
const formula::SingleVectorRefToken* pSVR = const formula::SingleVectorRefToken* pSVR =
dynamic_cast< const formula::SingleVectorRefToken* >(pCur); dynamic_cast< const formula::SingleVectorRefToken* >(pCur);
ss << "if (gid0 < " << pSVR->GetArrayLength() << "){\n "; ss << "if (gid0 < " << pSVR->GetArrayLength() << "){\n";
#else #else
nItems += 1;
#endif #endif
} }
else if (pCur->GetType() == formula::svDouble) else if (pCur->GetType() == formula::svDouble)
{ {
#ifdef ISNAN #ifdef ISNAN
ss << "{\n "; ss << "{\n";
#endif #endif
nItems += 1;
} }
else else
{ {
#ifdef ISNAN #ifdef ISNAN
#endif #endif
nItems += 1;
} }
#ifdef ISNAN #ifdef ISNAN
if(ocPush==vSubArguments[i]->GetFormulaToken()->GetOpCode()) if(ocPush==vSubArguments[i]->GetFormulaToken()->GetOpCode())
{ {
ss << "if (isNan("; ss <<" temp="<<vSubArguments[i]->GenSlidingWindowDeclRef();
ss << vSubArguments[i]->GenSlidingWindowDeclRef(); ss <<";\n";
ss << "))\n "; ss <<" if (isNan(temp))\n";
ss << "tmp"<<i<<"= 0;\n "; ss <<" tmp"<<i<<"= 0;\n";
ss << "else\n "; ss <<" else\n";
ss << "tmp"<<i<<"="<<vSubArguments[i]->GenSlidingWindowDeclRef(); ss <<" tmp"<<i<<"=temp;\n";
ss << ";\n }\n "; ss <<" }\n";
} }
else else
{ {
ss << "tmp"<<i<<"="<<vSubArguments[i]->GenSlidingWindowDeclRef(); ss <<" tmp"<<i<<"="<<vSubArguments[i]->GenSlidingWindowDeclRef(
ss <<";\n "; );
ss <<";\n";
} }
#else #else
ss << "tmp"<<i<<"="<<vSubArguments[i]->GenSlidingWindowDeclRef(); ss <<" tmp"<<i<<"="<<vSubArguments[i]->GenSlidingWindowDeclRef();
ss <<";\n "; ss <<";\n";
#endif #endif
} }
ss<<"if(tmp0==0.0)\n "; ss<<" if(tmp0==0.0)\n";
ss<<" return -(tmp2+tmp3)/tmp1;\n "; ss<<" return -(tmp2+tmp3)/tmp1;\n";
ss<<"double abl = pow(1.0+tmp0,tmp1);\n tmp-=tmp3"; ss<<" tmp-=tmp3;\n";
ss<<";\n tmp-=tmp2*abl;\n "; ss<<" tmp=tmp-tmp2*pow(1.0+tmp0,tmp1);\n";
ss<<"tmp =tmp/(1.0+tmp0*tmp4"; ss<<" tmp=tmp*pow(( (1.0+tmp0*tmp4)* ";
ss<<") / ( (abl-1.0)/tmp0);\n "; ss<<"( (pow(1.0+tmp0,tmp1)-1.0)/tmp0)),-1);\n";
ss << "return tmp;\n"; ss<<" return tmp;\n";
ss << "}"; ss<<"}";
} }
void OpNPV::GenSlidingWindowFunction(std::stringstream &ss, void OpNPV::GenSlidingWindowFunction(std::stringstream &ss,
const std::string sSymName, SubArguments &vSubArguments) const std::string sSymName, SubArguments &vSubArguments)
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment