blob: 198d0db24993b62c376ca5e9a9d95616f3257d37 [file] [log] [blame]
#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'");
}
}