mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-08-14 04:35:57 +08:00
Replace implicit cast with an explicit one
This commit is contained in:
parent
2f9401b061
commit
a071629fec
@ -503,7 +503,7 @@ struct TensorContractionEvaluatorBase
|
|||||||
|
|
||||||
// call gebp (matrix kernel)
|
// call gebp (matrix kernel)
|
||||||
// The parameters here are copied from Eigen's GEMM implementation
|
// The parameters here are copied from Eigen's GEMM implementation
|
||||||
gebp(output.getSubMapper(i2, j2), blockA, blockB, actual_mc, actual_kc, actual_nc, 1.0, -1, -1, 0, 0);
|
gebp(output.getSubMapper(i2, j2), blockA, blockB, actual_mc, actual_kc, actual_nc, Scalar(1), -1, -1, 0, 0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -376,7 +376,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
|
|||||||
const Index actual_mc = numext::mini(m_base_start + arg.mc, arg.max_m) - m_base_start;
|
const Index actual_mc = numext::mini(m_base_start + arg.mc, arg.max_m) - m_base_start;
|
||||||
gebp(arg.output.getSubMapper(m_base_start, arg.n),
|
gebp(arg.output.getSubMapper(m_base_start, arg.n),
|
||||||
(*arg.blockAs)[blockAId], arg.blockB,
|
(*arg.blockAs)[blockAId], arg.blockB,
|
||||||
actual_mc, arg.kc, arg.nc, 1.0, -1, -1, 0, 0);
|
actual_mc, arg.kc, arg.nc, Scalar(1), -1, -1, 0, 0);
|
||||||
|
|
||||||
// Notify that the kernel is done.
|
// Notify that the kernel is done.
|
||||||
const Index set_idx = blockAId * arg.n_blocks + arg.n_block_idx;
|
const Index set_idx = blockAId * arg.n_blocks + arg.n_block_idx;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user