added count_ones

LuisaGroup · Mar 25, 2024 · d90b260 · d90b260
1 parent a8692f2
commit d90b260
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/luisa_compute/src/lang/ops/impls.rs b/luisa_compute/src/lang/ops/impls.rs
@@ -176,6 +176,9 @@ where
     fn leading_zeros(&self) -> Expr<u32> {
         Func::Clz.call(self.clone())
     }
+    fn count_ones(&self) -> Expr<u32> {
+        Func::PopCount.call(self.clone())
+    }
 }
 
 macro_rules! impl_simple_fns {

diff --git a/luisa_compute/src/lang/ops/traits.rs b/luisa_compute/src/lang/ops/traits.rs
@@ -215,6 +215,7 @@ pub trait IntExpr {
     fn rotate_left(&self, n: Expr<u32>) -> Self;
     fn trailing_zeros(&self) -> Expr<u32>;
     fn leading_zeros(&self) -> Expr<u32>;
+    fn count_ones(&self) -> Expr<u32>;
 }
 
 pub trait FloatExpr: Sized {

diff --git a/luisa_compute_sys/LuisaCompute b/luisa_compute_sys/LuisaCompute
+8 −0		config/xmake_rules.lua
+33 −39		include/luisa/core/fiber.h
+20 −12		src/backends/cuda/cuda_builtin/cuda_device_math.h
+1,576 −1,531		src/backends/cuda/cuda_builtin_embedded.cpp
+1 −1		src/backends/cuda/cuda_builtin_embedded.h
+10 −4		src/backends/cuda/generate_device_library.py
+26 −10		src/core/xmake.lua
+7 −6		src/ext/CMakeLists.txt
+1 −1		src/ext/marl
+5 −5		src/rust/luisa_compute_backend_impl/src/cpu/codegen/cpu_prelude.h
+20 −12		src/rust/luisa_compute_backend_impl/src/cpu/codegen/device_math.h
+9 −28		src/tests/test_helloworld.cpp
+1 −1		src/xmake.lua