Remove duplicate macros
This commit is contained in:
parent
291125835e
commit
4644b47db1
@ -28,25 +28,6 @@ limitations under the License.
|
|||||||
#include "tensorflow/core/util/gpu_launch_config.h"
|
#include "tensorflow/core/util/gpu_launch_config.h"
|
||||||
#include "tensorflow/stream_executor/stream_executor.h"
|
#include "tensorflow/stream_executor/stream_executor.h"
|
||||||
|
|
||||||
#define TF_RETURN_IF_CUDA_ERROR(result) \
|
|
||||||
do { \
|
|
||||||
cudaError_t error(result); \
|
|
||||||
if (!SE_PREDICT_TRUE(error == cudaSuccess)) { \
|
|
||||||
return errors::Internal("Cuda call failed with ", \
|
|
||||||
cudaGetErrorString(error)); \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
#define TF_OP_REQUIRES_CUDA_SUCCESS(context, result) \
|
|
||||||
do { \
|
|
||||||
cudaError_t error(result); \
|
|
||||||
if (!SE_PREDICT_TRUE(error == cudaSuccess)) { \
|
|
||||||
context->SetStatus(errors::Internal("Cuda call failed with", \
|
|
||||||
cudaGetErrorString(error))); \
|
|
||||||
return; \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
struct __align__(16) Box {
|
struct __align__(16) Box {
|
||||||
float x1, y1, x2, y2;
|
float x1, y1, x2, y2;
|
||||||
};
|
};
|
||||||
|
Loading…
Reference in New Issue
Block a user