7f2a703ed5
* Merge with private repo * Update README * Update README * Update README * Add PyTorch requirements * Fix sync scopes for MQA logits (#256) * Update README
13 lines
276 B
C++
13 lines
276 B
C++
#pragma once
|
|
|
|
#include <optional>
|
|
#include <string>
|
|
|
|
namespace deep_gemm {
|
|
|
|
static std::string get_default_epilogue_type(const std::optional<std::string>& epilogue_type) {
|
|
return epilogue_type.value_or("epilogue::transform::EpilogueIdentity");
|
|
}
|
|
|
|
} // namespace deep_gemm
|