"docs/vscode:/vscode.git/clone" did not exist on "78e776a9ace5dc8007be43b536ad0f732688169d"
Unverified Commit 0d687968 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Add // NOLINT for the very long code. (#4834)



* alternative

* fix

* remove_todo

* blabl

* ablabl
Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent e3bf1825
...@@ -213,29 +213,23 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel( ...@@ -213,29 +213,23 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel(
// TODO(Steve): fix this long line in a separate PR. // TODO(Steve): fix this long line in a separate PR.
if (std::is_same<Op, op::CopyLhs<DType>>::value) { if (std::is_same<Op, op::CopyLhs<DType>>::value) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIDX_VECIN); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIDX_VECIN);
} else if (std::is_same<Op, op::CopyRhs<DType>>::value) { } else if (std::is_same<Op, op::CopyRhs<DType>>::value) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIN_VECIDX); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIN_VECIDX);
if (!has_idx) { if (!has_idx) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_IMPLICIT_INDEXED_VECIDX); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_IMPLICIT_INDEXED_VECIDX);
} }
} else { } else {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIDX_VECIN); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_OPORDER_VECIDX_VECIN);
if (has_idx) { if (has_idx) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_INDEXED_VEC); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_INDEXED_VEC);
} else { } else {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_IMPLICIT_INDEXED_VEC); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_IMPLICIT_INDEXED_VEC);
} }
} }
// Third, we set the Redop in the opredop_flags // Third, we set the Redop in the opredop_flags
...@@ -246,13 +240,11 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel( ...@@ -246,13 +240,11 @@ inline libxsmm_meltwfunction_opreduce_vecs_idx SpMMCreateLibxsmmKernel(
if (is_cmp) { if (is_cmp) {
if (Op::use_lhs) { if (Op::use_lhs) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_RECORD_ARGOP_OFF_VEC_0); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_RECORD_ARGOP_OFF_VEC_0);
} }
if (Op::use_rhs) { if (Op::use_rhs) {
opredop_flags = opredop_flags =
(libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | (libxsmm_meltw_opreduce_vecs_flags)(opredop_flags | LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_RECORD_ARGOP_OFF_VEC_1); // NOLINT
LIBXSMM_MELTW_FLAG_OPREDUCE_VECS_RECORD_ARGOP_OFF_VEC_1);
} }
} }
libxsmm_meltwfunction_opreduce_vecs_idx kernel = nullptr; libxsmm_meltwfunction_opreduce_vecs_idx kernel = nullptr;
...@@ -460,7 +452,7 @@ void SpMMRedopCsrOpt( ...@@ -460,7 +452,7 @@ void SpMMRedopCsrOpt(
IdType K_block_size = std::min( IdType K_block_size = std::min(
(int64_t)K, (int64_t)K,
(int64_t)(llc_size / (N * sizeof(DType) * nnz_prob * BLOCKING_HEURISTIC_PARAM))); (int64_t)(llc_size / (N * sizeof(DType) * nnz_prob * BLOCKING_HEURISTIC_PARAM))); // NOLINT
IdType M_block_size = M / (nthreads * NUM_BLOCKS_PER_THREAD); IdType M_block_size = M / (nthreads * NUM_BLOCKS_PER_THREAD);
if (M_block_size == 0) M_block_size = 1; if (M_block_size == 0) M_block_size = 1;
if (K_block_size == 0) K_block_size = 1; if (K_block_size == 0) K_block_size = 1;
......
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