issue/719 add more logs

include/infinicore/nn/module.hpp

@@ -6,6 +6,7 @@
 #include <type_traits>
 #include <unordered_map>
 #include <vector>
+#include <spdlog/spdlog.h>
 
 namespace infinicore::nn {
 class Module {

src/infinicore/utils.hpp

@@ -40,7 +40,10 @@ inline struct SpdlogInitializer {
             if (first_device___ != (tensor___)->device()) {                       \
                 throw std::runtime_error("Tensor devices mismatch "               \
                                          + first_device___.toString() + " vs "    \
-                                         + (tensor___)->device().toString() + "."); \
+                                         + (tensor___)->device().toString()       \
+                                         + " from " + std::string(__func__)       \
+                                         + " at " + std::string(__FILE__)         \
+                                         + ":" + std::to_string(__LINE__) + "."); \
             }                                                                     \
         }                                                                         \
     } while (0)