存图

     thread_local static bool save = false;
        thread_local static int count = 0;
        count++;
        if (!save && count > 100) {
            std::ofstream file("/ota/liuchenlong/" + cur_data_input_name +
                               ".bgr");
            char *data = NULL;
            if (cur_data_input_ptr->GetMemType() == MEM_DEVICE) {
                std::cout << "device atom" << std::endl;
                data = (char *)(cur_data_input_ptr->DevicePointer());
            } else {
                std::cout << "host atom" << std::endl;
                data = (char *)(cur_data_input_ptr->HostPointer());
            }
 
            file.write(data, 1024 * 576 * 3);
            save = true;
        }