@@ -44,14 +44,15 @@ bool shifted_;
44
44
val_->data <int8_t >() /* output*/ );
45
45
#else
46
46
typedef typename intgemm_<vtype>::type Integer;
47
+ auto input = child (0 )->val ();
47
48
if (!shifted_) {
48
- intgemm_<vtype>::width::PrepareA (child ( 0 )-> val () ->data (), /* input*/
49
+ intgemm_<vtype>::width::PrepareA (input ->data (), /* input*/
49
50
val_->data <Integer>(), /* output*/
50
51
*child (1 )->val ()->data (), /* Quant Mult*/
51
52
rows (child (0 )->val ()),
52
53
cols (child (0 )->val ()));
53
54
} else {
54
- intgemm::Int8Shift::PrepareA (child ( 0 )-> val () ->data (), /* input*/
55
+ intgemm::Int8Shift::PrepareA (input ->data (), /* input*/
55
56
val_->data <int8_t >(), /* output*/
56
57
*child (1 )->val ()->data (), /* Quant Mult*/
57
58
rows (child (0 )->val ()),
@@ -207,8 +208,8 @@ struct SelectColumnsBNodeOp : public UnaryNodeOp {
207
208
reinterpret_cast <Integer *>(input->data ()),
208
209
val_->data <Integer>(),
209
210
rows (input),
210
- &* indices_.begin (),
211
- &* indices_.end ());
211
+ indices_.data (),
212
+ indices_.data ()+indices_. size ());
212
213
#endif
213
214
}};
214
215
#else
@@ -274,7 +275,8 @@ struct QuantMultNodeOp : public UnaryNodeOp {
274
275
std::cerr << " Name: " << name () << " MeanAbs: " << meanstd.mean << " stddevAbs: " << meanstd.stddev << " Mean: " << meanstd2.mean << " stddev: "
275
276
<< meanstd2.stddev << " MaxAbs: " << intgemm::MaxAbsolute (child (0 )->val ()->data (), child (0 )->val ()->data () + child (0 )->val ()->shape ().elements ()) << std::endl;
276
277
}
277
- *val_->data () = 127 .0f / intgemm::MaxAbsolute (child (0 )->val ()->data (), child (0 )->val ()->data () + child (0 )->val ()->shape ().elements ());
278
+ auto input = child (0 )->val ();
279
+ *val_->data () = 127 .0f / intgemm::MaxAbsolute (input->data (), input->data () + input->size ());
278
280
}
279
281
)};
280
282
#else
0 commit comments