@@ -97,7 +97,8 @@ struct MarkCUDAOpsPass : public PassWrapper<MarkCUDAOpsPass, OperationPass<func:
9797 cols = 1 ;
9898 }
9999 logger->warn (" op: {}, opSize: {}, adding dims: {}x{}, bytes: {}\n setting unknowns to 1 for this test" ,
100- op->getName ().getStringRef ().str (), opSize, rows, cols, rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 );
100+ op->getName ().getStringRef ().str (), opSize, rows, cols,
101+ rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 );
101102 opSize += rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 ;
102103 }
103104 }
@@ -117,19 +118,15 @@ struct MarkCUDAOpsPass : public PassWrapper<MarkCUDAOpsPass, OperationPass<func:
117118 if (cols < 0 )
118119 cols = 1 ;
119120 }
120- logger->warn (" op: {}, opSize: {}, adding dims: {}x{}, bytes: {}\n " ,
121- op-> getName (). getStringRef (). str (), opSize, rows, cols, rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 );
121+ logger->warn (" op: {}, opSize: {}, adding dims: {}x{}, bytes: {}\n " , op-> getName (). getStringRef (). str (),
122+ opSize, rows, cols, rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 );
122123 opSize += rows * cols * t.getElementType ().getIntOrFloatBitWidth () / 8 ;
123124 }
124125 }
125- logger->debug (" Input size: {} KB, Output size: {} KB, Total required: {} MB, Mem budget: {}, required bytes: {}, budget bytes: {}" ,
126- inSize / 1024 ,
127- (opSize - inSize) / 1024 ,
128- opSize / 1048576 ,
129- mem_budget / 1048576 ,
130- opSize,
131- mem_budget
132- );
126+ logger->debug (" Input size: {} KB, Output size: {} KB, Total required: {} MB, Mem budget: {}, required bytes: "
127+ " {}, budget bytes: {}" ,
128+ inSize / 1024 , (opSize - inSize) / 1024 , opSize / 1048576 , mem_budget / 1048576 , opSize,
129+ mem_budget);
133130
134131 return opSize < mem_budget;
135132 }
0 commit comments