Evgeny Mankov
6acede94f5
Repeats clang's '--cuda-path' option.
[Reason]
In case of absence of any other clang's options setting '-cuda-path' allows not to specify separator '--' before clang's '--cuda-path'.
+ Tests and scripts are updated accordingly.
[ROCm/hip commit: fbfe005e4e]
19 rivejä
357 B
Plaintext
19 rivejä
357 B
Plaintext
// RUN: %run_test hipify "%s" "%t" %hipify_args %clang_args
|
|
|
|
#pragma once
|
|
|
|
#include <cuda_runtime.h>
|
|
|
|
|
|
/**
|
|
* Allocate GPU memory for `count` elements of type `T`.
|
|
*/
|
|
template<typename T>
|
|
static T* gpuMalloc(size_t count) {
|
|
T* ret = nullptr;
|
|
// CHECK: hipMalloc(&ret, count * sizeof(T));
|
|
cudaMalloc(&ret, count * sizeof(T));
|
|
return ret;
|
|
}
|
|
|