"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "65e6f48ce43fe21b0fb282ee8ece3421e1cd6e52"
Commit 3b5c6c7f authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 3eb036b2
...@@ -181,10 +181,8 @@ void miopen_gemm::fill_result(const shape& output_shape, ...@@ -181,10 +181,8 @@ void miopen_gemm::fill_result(const shape& output_shape,
{ {
output_shape.visit_type([&](auto as) { output_shape.visit_type([&](auto as) {
auto to_pointer = [&](auto&& arg) { return to_rocblas_type(as.from(arg.data())); }; auto to_pointer = [&](auto&& arg) { return to_rocblas_type(as.from(arg.data())); };
hipMemcpy(to_pointer(result), hipMemcpy(
to_pointer(c), to_pointer(result), to_pointer(c), output_shape.bytes(), hipMemcpyDeviceToDevice);
output_shape.bytes(),
hipMemcpyDeviceToDevice);
}); });
} }
else if(c.single()) else if(c.single())
...@@ -338,7 +336,8 @@ argument miopen_gemm::compute(context& ctx, ...@@ -338,7 +336,8 @@ argument miopen_gemm::compute(context& ctx,
to_pointer(args[1]), to_pointer(args[1]),
1, 1,
&beta_r, &beta_r,
to_pointer(args[2], batch_no * n), 1); to_pointer(args[2], batch_no * n),
1);
} }
}); });
} }
...@@ -376,7 +375,8 @@ argument miopen_gemm::compute(context& ctx, ...@@ -376,7 +375,8 @@ argument miopen_gemm::compute(context& ctx,
to_pointer(args[1], batch_no * m * n), to_pointer(args[1], batch_no * m * n),
1, 1,
&beta_r, &beta_r,
to_pointer(args[2], batch_no * m), 1); to_pointer(args[2], batch_no * m),
1);
} }
}); });
} }
......
...@@ -20,9 +20,7 @@ struct miopen_gemm ...@@ -20,9 +20,7 @@ struct miopen_gemm
int output_alias(const std::vector<shape>& shapes) const { return shapes.size() - 1; } int output_alias(const std::vector<shape>& shapes) const { return shapes.size() - 1; }
private: private:
void fill_result(const shape& output_shape, void fill_result(const shape& output_shape, const argument& result, const argument& c) const;
const argument& result,
const argument& c) const;
}; };
} // namespace gpu } // namespace gpu
......
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