Add missing device flags to some philox methods (#24)

msu-sparta · Dec 11, 2024 · ba9fec3 · ba9fec3
1 parent 92dc12a
commit ba9fec3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/openrand/philox.h b/include/openrand/philox.h
@@ -86,12 +86,12 @@ class Philox : public BaseRNG<Philox> {
     }
   }
 
-  openrand::uint4 draw_int4() {
+  OPENRAND_DEVICE openrand::uint4 draw_int4() {
     generate();
     return openrand::uint4{_out[0], _out[1], _out[2], _out[3]};
   }
 
-  openrand::float4 draw_float4() {
+  OPENRAND_DEVICE openrand::float4 draw_float4() {
     generate();
     return openrand::float4{
         u01<float, uint32_t>(_out[0]), u01<float, uint32_t>(_out[1]),