diff --git a/training/train.m b/training/train.m index 6fd4a86..6a70041 100644 --- a/training/train.m +++ b/training/train.m @@ -8,6 +8,8 @@ #include #include "backward.h" +int g_fp16_io = 0; // M1/M2: use fp16 I/O when cast op unsupported + static mach_timebase_info_data_t g_tb; static double ticksToMs(uint64_t t) { return (double)t * g_tb.numer / g_tb.denom / 1e6; }