Skip to content

Commit

Permalink
#13150: Cleanup LCM, GCD Macro (#13151)
Browse files Browse the repository at this point in the history
  • Loading branch information
VirdhatchaniKN authored Sep 26, 2024
1 parent 28e3825 commit 43eec24
Showing 1 changed file with 12 additions and 8 deletions.
20 changes: 12 additions & 8 deletions ttnn/cpp/ttnn/operations/eltwise/binary/binary_composite.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -201,15 +201,19 @@ struct ExecuteBinaryRemainder
const std::optional<MemoryConfig>& memory_config = std::nullopt);
};

#define DEFINE_BINARY_COMPOSITE(op_name) \
struct Execute##op_name { \
static Tensor invoke( \
const Tensor& input_tensor_a, \
const Tensor& input_tensor_b, \
const std::optional<MemoryConfig>& memory_config = std::nullopt); \
struct ExecuteLCM {
static Tensor invoke(
const Tensor& input_tensor_a,
const Tensor& input_tensor_b,
const std::optional<MemoryConfig>& memory_config = std::nullopt);
};

struct ExecuteGCD {
static Tensor invoke(
const Tensor& input_tensor_a,
const Tensor& input_tensor_b,
const std::optional<MemoryConfig>& memory_config = std::nullopt);
};
DEFINE_BINARY_COMPOSITE(LCM)
DEFINE_BINARY_COMPOSITE(GCD)

} // namespace binary
} // namespace operations
Expand Down

0 comments on commit 43eec24

Please sign in to comment.