host_guard.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdio>
#include <cstdlib>
 
#include "pulsatrix/device_backend.hpp"
 
#define PULSATRIX_REQUIRE_HOST(tensor)                                                              \
    do {                                                                                            \
        if ((tensor).device() != ::pulsatrix::DeviceType::Cpu) {                                    \
            std::fprintf(stderr,                                                                    \
                         "PULSATRIX_ASSERT failed: PULSATRIX_REQUIRE_HOST(%s): host-only code path "  \
                         "reached with a non-Cpu tensor at %s:%d\n",                                \
                         #tensor, __FILE__, __LINE__);                                              \
            std::abort();                                                                           \
        }                                                                                           \
    } while (0)