Skip to content
Snippets Groups Projects
Commit 27911af0 authored by kotsifa2's avatar kotsifa2
Browse files

Bug fix in VISCUtils.h, about generated metadata for cudnn target

parent 09b418d2
No related branches found
No related tags found
No related merge requests found
......@@ -159,6 +159,16 @@ void fixHintMetadata(Module &M, Function* F, Function* G) {
if(HintNode->getOperand(i) == MDT_F)
HintNode->setOperand(i, MDT_G);
}
HintNode = M.getOrInsertNamedMetadata("visc_hint_cudnn");
for(unsigned i = 0; i < HintNode->getNumOperands(); i++) {
if(HintNode->getOperand(i) == MDT_F)
HintNode->setOperand(i, MDT_G);
}
HintNode = M.getOrInsertNamedMetadata("visc_hint_promise");
for(unsigned i = 0; i < HintNode->getNumOperands(); i++) {
if(HintNode->getOperand(i) == MDT_F)
HintNode->setOperand(i, MDT_G);
}
HintNode = M.getOrInsertNamedMetadata("visc_hint_cpu");
for(unsigned i = 0; i < HintNode->getNumOperands(); i++) {
if(HintNode->getOperand(i) == MDT_F)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment