Skip to content

Commit 064df80

Browse files
committed
Fix compilation errors
1 parent 7e0467c commit 064df80

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

example/65_gemm_multiply_multiply/run_gemm_multiply_multiply_wp_example.inc

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -64,22 +64,22 @@ int run_gemm_example(int argc, char* argv[])
6464

6565
if(std::is_same<decltype(layout), ck::tensor_layout::gemm::RowMajor>::value)
6666
{
67-
return HostTensorDescriptor({row, col}, {stride, 1_uz});
67+
return ck::HostTensorDescriptor({row, col}, {stride, 1_uz});
6868
}
6969
else
7070
{
71-
return HostTensorDescriptor({row, col}, {1_uz, stride});
71+
return ck::HostTensorDescriptor({row, col}, {1_uz, stride});
7272
}
7373
};
7474

75-
Tensor<A0DataType> a0_m_k(f_host_tensor_descriptor(M, K, StrideA, A0Layout{}));
76-
Tensor<B0DataType> b0_k_n(f_host_tensor_descriptor(K, N, StrideB, B0Layout{}));
77-
Tensor<B0DataType> b0_preshuffled(
75+
ck::Tensor<A0DataType> a0_m_k(f_host_tensor_descriptor(M, K, StrideA, A0Layout{}));
76+
ck::Tensor<B0DataType> b0_k_n(f_host_tensor_descriptor(K, N, StrideB, B0Layout{}));
77+
ck::Tensor<B0DataType> b0_preshuffled(
7878
f_host_tensor_descriptor(K, N, StrideB, B0Layout{})); // use laout only for size
79-
Tensor<D0DataType> d0_m_n(f_host_tensor_descriptor(M, N, StrideD, D0Layout{}));
80-
Tensor<D1DataType> d1_m_n(f_host_tensor_descriptor(M, N, StrideD, D1Layout{}));
81-
Tensor<EDataType> e_m_n_host_result(f_host_tensor_descriptor(M, N, StrideE, ELayout{}));
82-
Tensor<EDataType> e_m_n_device_result(f_host_tensor_descriptor(M, N, StrideE, ELayout{}));
79+
ck::Tensor<D0DataType> d0_m_n(f_host_tensor_descriptor(M, N, StrideD, D0Layout{}));
80+
ck::Tensor<D1DataType> d1_m_n(f_host_tensor_descriptor(M, N, StrideD, D1Layout{}));
81+
ck::Tensor<EDataType> e_m_n_host_result(f_host_tensor_descriptor(M, N, StrideE, ELayout{}));
82+
ck::Tensor<EDataType> e_m_n_device_result(f_host_tensor_descriptor(M, N, StrideE, ELayout{}));
8383

8484
// Update strides based on tensor properties if they are <= 0
8585
auto get_stride = [](auto& tensor, auto layout, ck::index_t current_stride) -> ck::index_t {
@@ -130,11 +130,11 @@ int run_gemm_example(int argc, char* argv[])
130130
d0_m_n.GenerateTensorValue(GeneratorTensor_3<D0DataType>{0.0, 1.0});
131131
d1_m_n.GenerateTensorValue(GeneratorTensor_3<D1DataType>{0.0, 1.0});
132132
}
133-
DeviceMem a0_device_buf(sizeof(A0DataType) * a0_m_k.mDesc.GetElementSpaceSize());
134-
DeviceMem b0_device_buf(sizeof(B0DataType) * b0_k_n.mDesc.GetElementSpaceSize());
135-
DeviceMem d0_device_buf(sizeof(D0DataType) * d0_m_n.mDesc.GetElementSpaceSize());
136-
DeviceMem d1_device_buf(sizeof(D1DataType) * d1_m_n.mDesc.GetElementSpaceSize());
137-
DeviceMem e_device_buf(sizeof(EDataType) * e_m_n_device_result.mDesc.GetElementSpaceSize());
133+
ck::DeviceMem a0_device_buf(sizeof(A0DataType) * a0_m_k.mDesc.GetElementSpaceSize());
134+
ck::DeviceMem b0_device_buf(sizeof(B0DataType) * b0_k_n.mDesc.GetElementSpaceSize());
135+
ck::DeviceMem d0_device_buf(sizeof(D0DataType) * d0_m_n.mDesc.GetElementSpaceSize());
136+
ck::DeviceMem d1_device_buf(sizeof(D1DataType) * d1_m_n.mDesc.GetElementSpaceSize());
137+
ck::DeviceMem e_device_buf(sizeof(EDataType) * e_m_n_device_result.mDesc.GetElementSpaceSize());
138138

139139
a0_device_buf.ToDevice(a0_m_k.mData.data());
140140
d0_device_buf.ToDevice(d0_m_n.mData.data());
@@ -202,7 +202,7 @@ int run_gemm_example(int argc, char* argv[])
202202

203203
e_device_buf.FromDevice(e_m_n_device_result.mData.data());
204204

205-
Tensor<CShuffleDataType> c_m_n({M, N});
205+
ck::Tensor<CShuffleDataType> c_m_n({M, N});
206206

207207
using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<A0DataType,
208208
B0DataType,

0 commit comments

Comments
 (0)