CheckGenerator.h

#pragma once
#include "ATen/CPUGenerator.h"

namespace at {
static inline CPUGenerator * check_generator(Generator* expr) {
  if(auto result = dynamic_cast<CPUGenerator*>(expr))
    return result;
  runtime_error("Expected a 'CPUGenerator' but found 'CUDAGenerator'");
}
}