mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
17 lines
414 B
C++
17 lines
414 B
C++
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
#include <torch/torch.h>
|
||
|
|
#include <torch/fft.h>
|
||
|
|
|
||
|
|
#include <test/cpp/api/support.h>
|
||
|
|
|
||
|
|
// NOTE: Visual Studio and ROCm builds don't understand complex literals
|
||
|
|
// as of August 2020
|
||
|
|
|
||
|
|
// Simple test that verifies the fft namespace is registered properly
|
||
|
|
// properly in C++
|
||
|
|
TEST(FFTTest, fft) {
|
||
|
|
auto t = torch::randn(128, torch::dtype(torch::kComplexDouble));
|
||
|
|
torch::fft::fft(t);
|
||
|
|
}
|