"openmmapi/src/PythonForceImpl.cpp" did not exist on "0b662b4991e37286331dd4b5755d53f2b79fd1dd"
Commit b1ba831f authored by zhanghj2's avatar zhanghj2
Browse files

优化combine

parent 91691124
...@@ -191,7 +191,7 @@ flash_fwd_mla_combine_kernel(const CombineParams params) { ...@@ -191,7 +191,7 @@ flash_fwd_mla_combine_kernel(const CombineParams params) {
// return res; // return res;
// }; // };
// float4 data = result[i]; float4 data = result[i];
// o_ptr[i * 4] = float32_to_bfloat16((data.x)); // o_ptr[i * 4] = float32_to_bfloat16((data.x));
// o_ptr[i * 4 + 1] = float32_to_bfloat16((data.y)); // o_ptr[i * 4 + 1] = float32_to_bfloat16((data.y));
// o_ptr[i * 4 + 2] = float32_to_bfloat16((data.z)); // o_ptr[i * 4 + 2] = float32_to_bfloat16((data.z));
......
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