pytorch/test/cpp/api/special.cpp

16 lines
451 B
C++
Raw Normal View History

#include <gtest/gtest.h>
#include <torch/torch.h>
#include <torch/special.h>
#include <test/cpp/api/support.h>
// Simple test that verifies the special namespace is registered properly
// properly in C++
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
TEST(SpecialTest, special) {
// NOLINTNEXTLINE(cppcoreguidelines-avoid-magic-numbers)
auto t = torch::randn(128, torch::kDouble);
torch::special::gammaln(t);
}