avoid compiling for hip
@@ -265,6 +265,20 @@ if not use_hip:
'check_method': build.check_jitify_version,
'version_method': build.get_jitify_version,
})
+
+ MODULES.append({
+ 'name': 'random',
+ 'file': [
+ ('cupy.random._generator', ['cupy/random/cupy_distributions.cu']),
+ ],
+ 'include': [
+ ],
+ 'libraries': [
+ 'cudart',
+ 'curand',
+ ],
+ })
+
else:
MODULES.append({
'name': 'cub',
@@ -329,20 +343,6 @@ if bool(int(os.environ.get('CUPY_SETUP_ENABLE_THRUST', 1))):
})
-MODULES.append({
- 'name': 'random',
- 'file': [
- ('cupy.random._generator', ['cupy/random/cupy_distributions.cu']),
- ],
- 'include': [
- ],
- 'libraries': [
- 'cudart',
- 'curand',
- ],
-})
-
-
def ensure_module_file(file):
if isinstance(file, tuple):
return file
