Commit 68487b2a authored by yuguo's avatar yuguo
Browse files

[DCU] fix 2.5 compile issues

parent 25e709aa
......@@ -18,6 +18,21 @@
#include "cuda_runtime.h"
#include "logging.h"
static inline int getIntEnv(const char *name, int defval, int minval)
{
int val = defval;
const char* env = std::getenv(name);
if (env != nullptr && env[0] != '\0')
{
val = atoi(env);
if (val < minval)
{
val = minval;
}
}
return val;
}
namespace transformer_engine::detail {
cudaStream_t get_compute_stream(int idx) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment