mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +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
36 lines
863 B
C++
36 lines
863 B
C++
#include "test.h"
|
|
|
|
CASE("misc/no_grad/1") {
|
|
no_grad_guard guard;
|
|
auto model = Linear(5, 2).make();
|
|
auto x = Var(at::CPU(at::kFloat).randn({10, 5}), true);
|
|
auto y = model->forward({x})[0];
|
|
Variable s = y.sum();
|
|
|
|
backward(s);
|
|
EXPECT(!model->parameters()["weight"].grad().defined());
|
|
};
|
|
|
|
CASE("misc/random/seed_cpu") {
|
|
int size = 100;
|
|
setSeed(7);
|
|
auto x1 = Var(at::CPU(at::kFloat).randn({size}));
|
|
setSeed(7);
|
|
auto x2 = Var(at::CPU(at::kFloat).randn({size}));
|
|
|
|
auto l_inf = (x1.data() - x2.data()).abs().max().toCFloat();
|
|
EXPECT(l_inf < 1e-10);
|
|
};
|
|
|
|
CASE("misc/random/seed_cuda") {
|
|
CUDA_GUARD;
|
|
int size = 100;
|
|
setSeed(7);
|
|
auto x1 = Var(at::CUDA(at::kFloat).randn({size}));
|
|
setSeed(7);
|
|
auto x2 = Var(at::CUDA(at::kFloat).randn({size}));
|
|
|
|
auto l_inf = (x1.data() - x2.data()).abs().max().toCFloat();
|
|
EXPECT(l_inf < 1e-10);
|
|
};
|