sglang_v0.5.2/pytorch_2.8.0/test/cpp/api/special.cpp

14 lines
316 B
C++

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