Skip to content
Snippets Groups Projects
Commit 2c507d79 authored by Richard Angersbach's avatar Richard Angersbach
Browse files

Fix required headers for cuda/hip platforms

parent 16a6e80d
1 merge request!438Reduction Support
...@@ -8,4 +8,6 @@ class CudaPlatform(GenericGpu): ...@@ -8,4 +8,6 @@ class CudaPlatform(GenericGpu):
@property @property
def required_headers(self) -> set[str]: def required_headers(self) -> set[str]:
return set() return {
'"gpu_atomics.h"',
}
...@@ -8,4 +8,7 @@ class HipPlatform(GenericGpu): ...@@ -8,4 +8,7 @@ class HipPlatform(GenericGpu):
@property @property
def required_headers(self) -> set[str]: def required_headers(self) -> set[str]:
return {'"pystencils_runtime/hip.h"'} return {
'"gpu_atomics.h"',
'"pystencils_runtime/hip.h"',
}
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