".github/vscode:/vscode.git/clone" did not exist on "c2717317f03b12535cfd02b477ace61189e10e4b"
Commit 1e741191 authored by mtgu0705's avatar mtgu0705
Browse files

init method revert and reformat some files

parent 87ad5225
...@@ -217,8 +217,7 @@ int main(int argc, char* argv[]) ...@@ -217,8 +217,7 @@ int main(int argc, char* argv[])
a0_m_k.GenerateTensorValue(GeneratorTensor_1<A0DataType>{}); a0_m_k.GenerateTensorValue(GeneratorTensor_1<A0DataType>{});
quant_b0_k_n.GenerateTensorValue(GeneratorTensor_1<QuantDataType>{}); quant_b0_k_n.GenerateTensorValue(GeneratorTensor_1<QuantDataType>{});
// a1_m_k.GenerateTensorValue(GeneratorTensor_1<A1DataType>{}); // a1_m_k.GenerateTensorValue(GeneratorTensor_1<A1DataType>{});
// b1_k_n.GenerateTensorValue(GeneratorTensor_1<B1DataType>{}); b1_k_n.GenerateTensorValue(GeneratorTensor_1<B1DataType>{});
b1_k_n.GenerateTensorValue(GeneratorTensor_3<B1DataType>{0, 1.0});
break; break;
case 3: case 3:
a0_m_k.GenerateTensorValue(GeneratorTensor_2<A0DataType>{-2, 2}); a0_m_k.GenerateTensorValue(GeneratorTensor_2<A0DataType>{-2, 2});
......
...@@ -1306,28 +1306,28 @@ struct GridwiseGemmMultiD_BScale_xdl_cshuffle_v3 ...@@ -1306,28 +1306,28 @@ struct GridwiseGemmMultiD_BScale_xdl_cshuffle_v3
// B matrix blockwise copy // B matrix blockwise copy
auto b_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_b_scale< auto b_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_b_scale<
ThisThreadBlock, ThisThreadBlock,
BElementwiseOperation, BElementwiseOperation,
ck::tensor_operation::element_wise::PassThrough, ck::tensor_operation::element_wise::PassThrough,
InMemoryDataOperationEnum::Set, InMemoryDataOperationEnum::Set,
Sequence<BK0Number, NPerBlock, BK1Number>, Sequence<BK0Number, NPerBlock, BK1Number>,
BBlockTransferThreadClusterLengths_BK0_N_BK1, BBlockTransferThreadClusterLengths_BK0_N_BK1,
BBlockTransferThreadClusterArrangeOrder, BBlockTransferThreadClusterArrangeOrder,
BDataType, BDataType,
LDSTypeB, LDSTypeB,
decltype(b_grid_desc_bk0_n_bk1), decltype(b_grid_desc_bk0_n_bk1),
decltype(b_block_desc_bk0_n_bk1), decltype(b_block_desc_bk0_n_bk1),
BBlockTransferSrcAccessOrder, BBlockTransferSrcAccessOrder,
Sequence<0, 1, 2>, Sequence<0, 1, 2>,
BBlockTransferSrcVectorDim, BBlockTransferSrcVectorDim,
2, 2,
BBlockTransferSrcScalarPerVector, BBlockTransferSrcScalarPerVector,
BBlockTransferDstScalarPerVector_BK1, BBlockTransferDstScalarPerVector_BK1,
1, 1,
1, 1,
BThreadTransferSrcResetCoordinateAfterRun, BThreadTransferSrcResetCoordinateAfterRun,
true, true,
BlockwiseGemmPipe::GlobalBufferNum>(b_grid_desc_bk0_n_bk1, BlockwiseGemmPipe::GlobalBufferNum>(b_grid_desc_bk0_n_bk1,
make_multi_index(0, n_block_data_idx_on_grid, 0), make_multi_index(0, n_block_data_idx_on_grid, 0),
b_element_op, b_element_op,
b_block_desc_bk0_n_bk1, b_block_desc_bk0_n_bk1,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment