mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
af71fb882f
* Dump autogradpp into PyTorch * Fixed up CMake for autogradpp/C++ API * Made cereal a submodule * Change search location of autogradpps mnist directory * Add test_api to CI * Download MNIST from the internet instead of storing in repo * Fix warnings
15 lines
289 B
C++
15 lines
289 B
C++
#pragma once
|
|
|
|
#include "lest.hpp"
|
|
#include <torch/autograd.h>
|
|
|
|
using namespace autograd;
|
|
|
|
#define CASE( name ) lest_CASE( specification(), name )
|
|
|
|
#define CUDA_GUARD if (!hasCuda()) {\
|
|
std::cerr << "No cuda, skipping test" << std::endl; return;\
|
|
}
|
|
|
|
extern lest::tests & specification();
|