Review: Add __restrict__ keywords

Review: Add restrict keywords
1e012e18 · Bartlomiej Wroblewski · 6ac56691 · 1e012e18
Commit 1e012e18 authored Aug 31, 2023 by Bartlomiej Wroblewski
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

include/ck/tensor_operation/gpu/grid/gridwise_gemm_dpp.hpp include/ck/tensor_operation/gpu/grid/gridwise_gemm_dpp.hpp +3 -3

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_dpp.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_dpp.hpp
@@ -466,9 +466,9 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_dpp
              typename AGridDesc_K0_M_K1,
              typename BGridDesc_K0_N_K1,
              typename CGridDesc_M_N>
-    __device__ static void Run(const FloatAB* p_a_grid,
-                               const FloatAB* p_b_grid,
-                               FloatC* p_c_grid,
+    __device__ static void Run(const FloatAB* __restrict__ p_a_grid,
+                               const FloatAB* __restrict__ p_b_grid,
+                               FloatC* __restrict__ p_c_grid,
                               void* __restrict__ p_shared,
                               const AGridDesc_K0_M_K1& a_grid_desc_k0_m_k1,
                               const BGridDesc_K0_N_K1& b_grid_desc_k0_n_k1,