warp-lang 1.1.0__py3-none-manylinux2014_x86_64.whl → 1.2.0__py3-none-manylinux2014_x86_64.whl

warp/native/quat.h

@@ -56,6 +56,33 @@ inline bool CUDA_CALLABLE isfinite(const quat_t<Type>& q)
     return isfinite(q.x) && isfinite(q.y) && isfinite(q.z) && isfinite(q.w);
 }
 
+template<typename Type>
+inline void CUDA_CALLABLE adj_isfinite(const quat_t<Type>& q, quat_t<Type>& adj_q, const bool &adj_ret)
+{
+}
+
+template<typename Type>
+inline bool CUDA_CALLABLE isnan(const quat_t<Type>& q)
+{
+    return isnan(q.x) || isnan(q.y) || isnan(q.z) || isnan(q.w);
+}
+
+template<typename Type>
+inline void CUDA_CALLABLE adj_isnan(const quat_t<Type>& q, quat_t<Type>& adj_q, const bool &adj_ret)
+{
+}
+
+template<typename Type>
+inline bool CUDA_CALLABLE isinf(const quat_t<Type>& q)
+{
+    return isinf(q.x) || isinf(q.y) || isinf(q.z) || isinf(q.w);
+}
+
+template<typename Type>
+inline void CUDA_CALLABLE adj_isinf(const quat_t<Type>& q, quat_t<Type>& adj_q, const bool &adj_ret)
+{
+}
+
 template<typename Type>
 inline CUDA_CALLABLE quat_t<Type> atomic_add(quat_t<Type> * addr, quat_t<Type> value) 
 {
@@ -1082,4 +1109,4 @@ inline CUDA_CALLABLE quat_t<Type> quat_identity()
 
 
 
-} // namespace wp
+} // namespace wp

warp/native/rand.h

@@ -13,6 +13,10 @@
 #define M_PI_F 3.14159265358979323846f
 #endif
 
+#ifndef LOG_EPSILON
+#define LOG_EPSILON 5.96e-8f
+#endif
+
 namespace wp
 {
 
@@ -33,7 +37,7 @@ inline CUDA_CALLABLE float randf(uint32& state) { state = rand_pcg(state); retur
 inline CUDA_CALLABLE float randf(uint32& state, float min, float max) { return (max - min) * randf(state) + min; }
 
 // Box-Muller method
-inline CUDA_CALLABLE float randn(uint32& state) { return sqrt(-2.f * log(randf(state))) * cos(2.f * M_PI_F * randf(state)); }
+inline CUDA_CALLABLE float randn(uint32& state) { return sqrt(-2.f * log(randf(state) + LOG_EPSILON)) * cos(2.f * M_PI_F * randf(state)); }
 
 inline CUDA_CALLABLE void adj_rand_init(int seed, int& adj_seed, float adj_ret) {}
 inline CUDA_CALLABLE void adj_rand_init(int seed, int offset, int& adj_seed, int& adj_offset, float adj_ret) {}

warp/native/vec.h

@@ -551,6 +551,32 @@ inline bool CUDA_CALLABLE isfinite(vec_t<Length, Type> x)
     return true;
 }
 
+template<unsigned Length, typename Type>
+inline bool CUDA_CALLABLE isnan(vec_t<Length, Type> x)
+{
+    for( unsigned i=0; i < Length; ++i )
+    {
+        if(isnan(x[i]))
+        {
+            return true;
+        }
+    }
+    return false;
+}
+
+template<unsigned Length, typename Type>
+inline bool CUDA_CALLABLE isinf(vec_t<Length, Type> x)
+{
+    for( unsigned i=0; i < Length; ++i )
+    {
+        if(isinf(x[i]))
+        {
+            return true;
+        }
+    }
+    return false;
+}
+
 // These two functions seem to compile very slowly
 template<unsigned Length, typename Type>
 inline CUDA_CALLABLE vec_t<Length,Type> min(vec_t<Length,Type> a, vec_t<Length,Type> b)
@@ -964,6 +990,24 @@ inline CUDA_CALLABLE void adj_cross(vec_t<3,Type> a, vec_t<3,Type> b, vec_t<3,Ty
     adj_b -= cross(a, adj_ret);
 }
 
+template<unsigned Length, typename Type>
+inline CUDA_CALLABLE void adj_isfinite(const vec_t<Length, Type> &x, vec_t<Length,Type>& adj_x, const bool &adj_ret)
+{
+
+}
+
+template<unsigned Length, typename Type>
+inline CUDA_CALLABLE void adj_isnan(const vec_t<Length, Type> &x, vec_t<Length,Type>& adj_x, const bool &adj_ret)
+{
+
+}
+
+template<unsigned Length, typename Type>
+inline CUDA_CALLABLE void adj_isinf(const vec_t<Length, Type> &x, vec_t<Length,Type>& adj_x, const bool &adj_ret)
+{
+
+}
+
 template<unsigned Length, typename Type>
 inline CUDA_CALLABLE void adj_min(const vec_t<Length,Type> &a, const vec_t<Length,Type> &b, vec_t<Length,Type>& adj_a, vec_t<Length,Type>& adj_b, const vec_t<Length,Type> &adj_ret)
 {
@@ -1130,4 +1174,4 @@ inline CUDA_CALLABLE void adj_vec4(float s, float& adj_s, const vec4& adj_ret)
 }
 
 
-} // namespace wp
+} // namespace wp