mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-06-10 03:23:29 +03:00
a71f07694c
- Make passthru.tests an attrset. - Remove CMake VERBOSE option. - Remove spurious `cat CMakeLists.txt`. - Run the test as well. - Actually test some functionality.
20 lines
301 B
C++
20 lines
301 B
C++
#undef NDEBUG
|
|
#include <cassert>
|
|
|
|
#include <iostream>
|
|
|
|
#include <torch/torch.h>
|
|
|
|
int main() {
|
|
torch::Tensor tensor = torch::eye(3);
|
|
|
|
float checkData[] = {
|
|
1, 0, 0,
|
|
0, 1, 0,
|
|
0, 0, 1
|
|
};
|
|
|
|
torch::Tensor check = torch::from_blob(checkData, {3, 3});
|
|
|
|
assert(tensor.allclose(check));
|
|
}
|