MCPcopy Create free account
hub / github.com/MegEngine/MegCC / Rdeucer

Method Rdeucer

compiler/lib/KernelGen/GeneralIntrinsic/Reduce.cpp:16–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

14 std::string load_str;
15 std::string store_str;
16 Rdeucer(std::string ctype) : m_ctype(ctype) {
17 std::string suffix_ins;
18 if (m_ctype == "float") {
19 simd_type = "GI_FLOAT32_t";
20 suffix_ins = "Float32";
21 } else if (m_ctype == "gi_float16_t") {
22 simd_type = "GI_FLOAT16_t";
23 suffix_ins = "Float16";
24 } else {
25 CC_ABORT << "unsupported dtype for min reduce";
26 }
27 init_str = "GiBroadcast" + suffix_ins;
28 load_str = "GiLoad" + suffix_ins;
29 store_str = "GiStore" + suffix_ins;
30 }
31 virtual std::string gen_init(bool c_remain) = 0;
32 virtual std::string gen_feed() = 0;
33 virtual std::string gen_feed_remain() = 0;

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected