ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add default arguments for prologue and epilogue. (#2020)

Browse Source
This commit is contained in:
Mirza Halilčević
2025-03-26 17:28:40 +01:00
committed by GitHub
parent 99b2bbc1d6
commit 21e0ca197d
4 changed files with 17 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
codegen/include/ck/host/device_batched_gemm_softmax_gemm/problem.hpp
View File

@@ -39,8 +39,8 @@ struct Problem
// returns a list of instances based on the problem spec and provided fusion operations
std::vector<Solution> GetSolutions(const std::string& arch,
const std::string& prologue,
const std::string& epilogue) const;
const std::string& prologue = "",
const std::string& epilogue = "") const;
};
} // namespace device_batched_gemm_softmax_gemm

4
codegen/include/ck/host/device_gemm_multiple_d/problem.hpp
View File

@@ -37,8 +37,8 @@ struct Problem
// returns a list of instances based on the problem spec and provided fusion operations
std::vector<Solution> GetSolutions(const std::string& arch,
const std::string& prologue,
const std::string& epilogue) const;
const std::string& prologue = "",
const std::string& epilogue = "") const;
};
} // namespace device_gemm_multiple_d