libtorch-bin: cleanup passthru test

- Make passthru.tests an attrset.
- Remove CMake VERBOSE option.
- Remove spurious `cat CMakeLists.txt`.
- Run the test as well.
- Actually test some functionality.
This commit is contained in:
Daniël de Kok 2020-11-07 13:50:23 +01:00
parent 126e5c1bd1
commit a71f07694c
3 changed files with 18 additions and 9 deletions

View File

@ -100,7 +100,7 @@ in stdenv.mkDerivation {
outputs = [ "out" "dev" ];
passthru.tests = callPackage ./test { };
passthru.tests.cmake = callPackage ./test { };
meta = with stdenv.lib; {
description = "C++ API of the PyTorch machine learning framework";

View File

@ -6,16 +6,12 @@ stdenv.mkDerivation {
src = ./.;
postPatch = ''
cat CMakeLists.txt
'';
makeFlags = [ "VERBOSE=1" ];
nativeBuildInputs = [ cmake ];
buildInputs = [ libtorch-bin ];
doCheck = true;
installPhase = ''
touch $out
'';

View File

@ -1,7 +1,20 @@
#include <torch/torch.h>
#undef NDEBUG
#include <cassert>
#include <iostream>
#include <torch/torch.h>
int main() {
torch::Tensor tensor = torch::eye(3);
std::cout << tensor << std::endl;
float checkData[] = {
1, 0, 0,
0, 1, 0,
0, 0, 1
};
torch::Tensor check = torch::from_blob(checkData, {3, 3});
assert(tensor.allclose(check));
}