diff options
| author | Xiaomeng Yang <yangxm@fb.com> | 2019-02-07 18:19:46 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-02-07 18:38:26 -0800 |
| commit | 2db847b3a7edc48652e144e7c9d7aa0bbed66aaa (patch) | |
| tree | 3d95b2f625ca878f038092dea671a2bffed088c6 /caffe2/utils/math_gpu_test.cc | |
| parent | 22477c6a7fc327cdf913751ab31b788ec710caa9 (diff) | |
| download | pytorch-2db847b3a7edc48652e144e7c9d7aa0bbed66aaa.tar.gz pytorch-2db847b3a7edc48652e144e7c9d7aa0bbed66aaa.tar.bz2 pytorch-2db847b3a7edc48652e144e7c9d7aa0bbed66aaa.zip | |
Separate elementwise level2 math functions (#16753)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16753
Separate elementwise level2 math functions
i-am-not-moving-c2-to-c10
Reviewed By: houseroad
Differential Revision: D13954928
fbshipit-source-id: 1ca7a5d3da96e32510f502e5e4e79168854bee67
Diffstat (limited to 'caffe2/utils/math_gpu_test.cc')
| -rw-r--r-- | caffe2/utils/math_gpu_test.cc | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/caffe2/utils/math_gpu_test.cc b/caffe2/utils/math_gpu_test.cc index 80d7ff0291..99e9f69e33 100644 --- a/caffe2/utils/math_gpu_test.cc +++ b/caffe2/utils/math_gpu_test.cc @@ -204,24 +204,6 @@ TEST(MathUtilGPUTest, testReduceMax) { [](int /*i*/) { return 17.0f; }); } -TEST(MathUtilGPUTest, testElemwiseMax) { - executeGpuBinaryOpTest( - 13, - 13, - 13, - [](int i) { return 2.0f - i; }, - [](int i) { return i - 6.0f; }, - [](int N0, - int /*N1*/, - const float* src0, - const float* src1, - float* dst, - CUDAContext* context) { - math::ElemwiseMax<float, CUDAContext>(N0, src0, src1, dst, context); - }, - [](int i) { return std::max(2.0f - i, i - 6.0f); }); -} - TEST(MathUtilGPUTest, testCopyVector) { executeGpuBinaryOpTest( 6, |