mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
30fb2c4aba
Let's have some fun. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78828 Approved by: https://github.com/ezyang
14 lines
316 B
C++
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);
|
|
}
|