"vscode:/vscode.git/clone" did not exist on "1066fe4cbc24ba21f9d0dec7886f35bb91afa281"
Commit bdf6cddb authored by root's avatar root
Browse files

clean up

parent 1bc44df2
...@@ -79,7 +79,7 @@ struct PassThroughPack2 ...@@ -79,7 +79,7 @@ struct PassThroughPack2
__host__ __device__ constexpr void operator()(ck::bhalf2_t& y, const ck::int8x2_t& x) const __host__ __device__ constexpr void operator()(ck::bhalf2_t& y, const ck::int8x2_t& x) const
{ {
y = ck::bit_cast<ck::bhalf2_t>(static_cast<int32_t>(ck::bit_cast<int16_t>(x))); y = ck::bit_cast<ck::bhalf2_t>(static_cast<int32_t>(ck::bit_cast<int16_t>(x)));
} }
template <> template <>
...@@ -91,12 +91,10 @@ struct PassThroughPack2 ...@@ -91,12 +91,10 @@ struct PassThroughPack2
constexpr const static bool is_pack2_invocable = true; constexpr const static bool is_pack2_invocable = true;
}; };
using PassThrough = ck::tensor_operation::element_wise::PassThrough; using PassThrough = ck::tensor_operation::element_wise::PassThrough;
using AddFastGelu = ck::tensor_operation::element_wise::AddFastGelu;
using AElementOp = PassThrough; using AElementOp = PassThrough;
using BElementOp = PassThroughPack2; using BElementOp = PassThrough;
using CDEElementOp = MultiplyAddFastGelu; using CDEElementOp = MultiplyAddFastGelu;
static constexpr auto GemmSpec = ck::tensor_operation::device::GemmSpecialization::Default; static constexpr auto GemmSpec = ck::tensor_operation::device::GemmSpecialization::Default;
......
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