Replace get_ptr() in the accessor with atomicAdd()

nipunG314 · nipunG314 · commit f7d0cc3a849c · 2024-05-25T19:05:03.000+05:30
diff --git a/CountingSort/app_resources/prefix_sum_shader.comp.hlsl b/CountingSort/app_resources/prefix_sum_shader.comp.hlsl
@@ -25,10 +25,12 @@ struct PtrAccessor
         deref().store(value);
     }
 
-    nbl::hlsl::bda::__spv_ptr_t<uint32_t> get_ptr(uint64_t index)
+    uint32_t atomicAdd(uint64_t index, uint32_t value)
     {
-        return nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template
+        nbl::hlsl::bda::__spv_ptr_t < uint32_t > ptr = nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template
         deref().get_ptr();
+
+        return nbl::hlsl::glsl::atomicAdd(ptr, value);
     }
 
     uint64_t addr;
diff --git a/CountingSort/app_resources/scatter_shader.comp.hlsl b/CountingSort/app_resources/scatter_shader.comp.hlsl
@@ -25,10 +25,12 @@ struct PtrAccessor
         deref().store(value);
     }
 
-    nbl::hlsl::bda::__spv_ptr_t<uint32_t> get_ptr(uint64_t index)
+    uint32_t atomicAdd(uint64_t index, uint32_t value)
     {
-        return nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template
+        nbl::hlsl::bda::__spv_ptr_t < uint32_t > ptr = nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template
         deref().get_ptr();
+
+        return nbl::hlsl::glsl::atomicAdd(ptr, value);
     }
 
     uint64_t addr;

Original file line number	Diff line number	Diff line change
`@@ -25,10 +25,12 @@ struct PtrAccessor`
`25`	`25`	`deref().store(value);`
`26`	`26`	`}`
`27`	`27`
`28`		`- nbl::hlsl::bda::__spv_ptr_t<uint32_t> get_ptr(uint64_t index)`
	`28`	`+ uint32_t atomicAdd(uint64_t index, uint32_t value)`
`29`	`29`	`{`
`30`		`- return nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template`
	`30`	`+ nbl::hlsl::bda::__spv_ptr_t < uint32_t > ptr = nbl::hlsl::bda::__ptr < uint32_t > (addr + sizeof(uint32_t) * index).template`
`31`	`31`	`deref().get_ptr();`
	`32`	`+`
	`33`	`+ return nbl::hlsl::glsl::atomicAdd(ptr, value);`
`32`	`34`	`}`
`33`	`35`
`34`	`36`	`uint64_t addr;`