[FORK][FEATURE] Add jit debug trace log dump in GCC debug mode

ONEDNN 3.2 migration squashed commits:
    - enable jit debug_trace() only when ONEDNN_JIT_DUMP is properly exported

ONEDNN 3.5 squash list:
[FIX]Don't use backtrace under Android

Author: usstq

src/cpu/x64/jit_generator.hpp

@@ -2791,7 +2791,31 @@ class jit_generator : public Xbyak::MmapAllocator,
     virtual const char *name() const = 0;
     virtual const char *source_file() const = 0;
 
+    void dump_debug_traces(const uint8_t *code, size_t code_size) const {
+#if !defined(NDEBUG) && defined(__GNUC__)
+        if (code && get_jit_dump()) {
+        #define MAX_FNAME_LEN 256
+            static int counter = 0;
+            char fname[MAX_FNAME_LEN + 1];
+            snprintf(fname, MAX_FNAME_LEN, "dnnl_traces_cpu_%s.%d.txt", name(),
+                    counter);
+            counter++;
+
+            std::cout << "[ oneDNN ] dump_debug_traces: " << fname << std::endl;
+            FILE *fp = fopen(fname, "wb+");
+            if (fp) {
+                for (const auto & p : get_debug_traces()) {
+                    fprintf(fp, "%lx:\t%s\n", p.first, p.second.c_str());
+                }
+                fclose(fp);
+            }
+        #undef MAX_FNAME_LEN
+        }
+#endif
+    }
+
     void register_jit_code(const Xbyak::uint8 *code, size_t code_size) const {
+        dump_debug_traces(code, code_size);
         jit_utils::register_jit_code(code, code_size, name(), source_file());
     }
 

src/cpu/x64/xbyak/xbyak.h

@@ -67,6 +67,13 @@
 #include <iostream>
 #endif
 
+// for debug trace in GCC
+#if !defined(NDEBUG) && defined(__GNUC__) && !(defined(__ANDROID__) || defined(ANDROID))
+#include <execinfo.h>
+#include <map>
+#include <sstream>
+#endif
+
 // #define XBYAK_DISABLE_AVX512
 
 #if !defined(XBYAK_USE_MMAP_ALLOCATOR) && !defined(XBYAK_DONT_USE_MMAP_ALLOCATOR)
@@ -2097,12 +2104,14 @@ class CodeGenerator : public CodeArray {
 	}
 	void opRR(const Reg& reg1, const Reg& reg2, uint64_t type, int code)
 	{
+        debug_trace();
 		bool rex2 = rex(reg2, reg1, type);
 		writeCode(type, reg1, code, rex2);
 		setModRM(3, reg1.getIdx(), reg2.getIdx());
 	}
 	void opMR(const Address& addr, const Reg& r, uint64_t type, int code, uint64_t type2 = 0, int code2 = NONE)
 	{
+        debug_trace();
 		if (code2 == NONE) code2 = code;
 		if (type2 && opROO(Reg(), addr, r, type2, code2)) return;
 		if (addr.is64bitDisp()) XBYAK_THROW(ERR_CANT_USE_64BIT_DISP)
@@ -2404,6 +2413,7 @@ class CodeGenerator : public CodeArray {
 		if (bit == 16 || bit == BIT) {
 			if (bit == 16) db(0x66);
 			if (op.isREG()) {
+				debug_trace();
 				if (op.getReg().getIdx() >= 8) db(0x41);
 				db(alt | (op.getIdx() & 7));
 				return;
@@ -2494,6 +2504,7 @@ class CodeGenerator : public CodeArray {
 	}
 	void opVex(const Reg& r, const Operand *p1, const Operand& op2, uint64_t type, int code, int imm8 = NONE)
 	{
+		debug_trace();
 		if (op2.isMEM()) {
 			Address addr = op2.getAddress();
 			const RegExp& regExp = addr.getRegExp();
@@ -3047,6 +3058,35 @@ class CodeGenerator : public CodeArray {
 		opRO(static_cast<const Reg&>(*p1), *p2, 0, 0x86 | (p1->isBit(8) ? 0 : 1), (p1->isREG() && (p1->getBit() == p2->getBit())));
 	}
 
+#if !defined(NDEBUG) && defined(__GNUC__) && !(defined(__ANDROID__) || defined(ANDROID))
+	std::map<size_t, std::string> debug_traces;
+	void debug_trace() {
+		static bool enable_trace = std::getenv("ONEDNN_JIT_DUMP") && atoi(std::getenv("ONEDNN_JIT_DUMP")) != 0;
+		if (!enable_trace)
+			return;
+
+		void *array[8];
+		int size = backtrace(array, 8);
+		char **strings = backtrace_symbols(array, size);
+		std::stringstream ss;
+		// skip first 2 frame (backtrace,trace_code)
+		for(int i = 2; i < size; ++i) ss << "," << strings[i];
+		auto offset = getSize();
+		if (debug_traces.count(offset)) {
+			debug_traces[offset] += ss.str();
+		} else {
+			debug_traces[offset] = ss.str();
+		}
+		free(strings);
+	}
+	std::map<size_t, std::string> get_debug_traces() const {
+		return debug_traces;
+	}
+#else
+	void debug_trace() {}
+	std::map<size_t, std::string> get_debug_traces() const { return {}; }
+#endif
+
 #ifndef XBYAK_DISABLE_SEGMENT
 	void push(const Segment& seg)
 	{