Fix compilation error due to interface change

This commit is contained in:
Po-Yen, Chen
2022-07-06 07:24:42 +08:00
parent 2e7b3899ea
commit 4017947895

View File

@@ -616,8 +616,7 @@ struct DeviceGemm_Xdl_ProducerConsumer_CShuffle : public DeviceGemm<ALayout,
index_t StrideC,
AElementwiseOperation a_element_op,
BElementwiseOperation b_element_op,
CElementwiseOperation c_element_op,
index_t /* KBatch */ = 1) override
CElementwiseOperation c_element_op) override
{
return std::make_unique<Argument>(static_cast<const ADataType*>(p_a),
static_cast<const BDataType*>(p_b),