Spaces:

qbhf2
/

GarmentCode

Paused

GarmentCode / NvidiaWarp-GarmentCode /warp /native /cutlass /examples /common /helper.h

added NvidiaWarp and GarmentCode repos

66c9c8a over 1 year ago

1.43 kB

#pragma once

#include "cuda_runtime.h"

#define CUTLASS_CHECK(status)                                                                    \
  {                                                                                              \
    cutlass::Status error = status;                                                              \
    if (error != cutlass::Status::kSuccess) {                                                    \
      std::cerr << "Got cutlass error: " << cutlassGetStatusString(error) << " at: " << __LINE__ \
                << std::endl;                                                                    \
      exit(EXIT_FAILURE);                                                                        \
    }                                                                                            \
  }

#define CUDA_CHECK(status)                                              \
  {                                                                     \
    cudaError_t error = status;                                         \
    if (error != cudaSuccess) {                                         \
      std::cerr << "Got bad cuda status: " << cudaGetErrorString(error) \
                << " at line: " << __LINE__ << std::endl;               \
      exit(EXIT_FAILURE);                                               \
    }                                                                   \
  }

	#pragma once

	#include "cuda_runtime.h"

	#define CUTLASS_CHECK(status) \
	{ \
	cutlass::Status error = status; \
	if (error != cutlass::Status::kSuccess) { \
	std::cerr << "Got cutlass error: " << cutlassGetStatusString(error) << " at: " << __LINE__ \
	<< std::endl; \
	exit(EXIT_FAILURE); \
	} \
	}

	#define CUDA_CHECK(status) \
	{ \
	cudaError_t error = status; \
	if (error != cudaSuccess) { \
	std::cerr << "Got bad cuda status: " << cudaGetErrorString(error) \
	<< " at line: " << __LINE__ << std::endl; \
	exit(EXIT_FAILURE); \
	} \
	}