mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a9b0a921d5
Summary: As GoogleTest `TEST` macro is non-compliant with it as well as `DEFINE_DISPATCH` All changes but the ones to `.clang-tidy` are generated using following script: ``` for i in `find . -type f -iname "*.c*" -or -iname "*.h"|xargs grep cppcoreguidelines-avoid-non-const-global-variables|cut -f1 -d:|sort|uniq`; do sed -i "/\/\/ NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)/d" $i; done ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/62008 Reviewed By: driazati, r-barnes Differential Revision: D29838584 Pulled By: malfet fbshipit-source-id: 1b2f8602c945bd4ce50a9bfdd204755556e31d13
42 lines
967 B
C++
42 lines
967 B
C++
#include <gtest/gtest.h>
|
|
#include "caffe2/core/context.h"
|
|
#include "caffe2/core/event.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
TEST(EventCPUTest, EventBasics) {
|
|
DeviceOption device_option;
|
|
device_option.set_device_type(PROTO_CPU);
|
|
Event event(device_option);
|
|
CPUContext context;
|
|
|
|
context.Record(&event);
|
|
event.SetFinished();
|
|
|
|
context.WaitEvent(event);
|
|
event.Finish();
|
|
|
|
event.Reset();
|
|
event.Record(CPU, &context);
|
|
event.SetFinished();
|
|
event.Wait(CPU, &context);
|
|
}
|
|
|
|
TEST(EventCPUTest, EventErrors) {
|
|
DeviceOption device_option;
|
|
device_option.set_device_type(PROTO_CPU);
|
|
Event event(device_option);
|
|
|
|
event.SetFinished();
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
ASSERT_THROW(event.SetFinished("error"), caffe2::EnforceNotMet);
|
|
ASSERT_EQ(event.ErrorMessage(), "No error");
|
|
|
|
event.Reset();
|
|
event.SetFinished("error 1");
|
|
event.SetFinished("error 2");
|
|
ASSERT_EQ(event.ErrorMessage(), "error 1");
|
|
}
|
|
|
|
} // namespace caffe2
|