warp-lang 1.0.0b5__py3-none-macosx_10_13_universal2.whl → 1.0.0b6__py3-none-macosx_10_13_universal2.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of warp-lang might be problematic. Click here for more details.

Files changed (188) hide show
  1. docs/conf.py +3 -4
  2. examples/env/env_ant.py +1 -1
  3. examples/env/env_cartpole.py +1 -1
  4. examples/env/env_humanoid.py +1 -1
  5. examples/example_dem.py +28 -26
  6. examples/example_diffray.py +37 -30
  7. examples/example_fluid.py +7 -3
  8. examples/example_jacobian_ik.py +1 -1
  9. examples/example_mesh_intersect.py +10 -7
  10. examples/example_nvdb.py +3 -3
  11. examples/example_render_opengl.py +19 -10
  12. examples/example_sim_cartpole.py +9 -5
  13. examples/example_sim_cloth.py +29 -25
  14. examples/example_sim_fk_grad.py +2 -2
  15. examples/example_sim_fk_grad_torch.py +3 -3
  16. examples/example_sim_grad_bounce.py +11 -8
  17. examples/example_sim_grad_cloth.py +12 -9
  18. examples/example_sim_granular.py +2 -2
  19. examples/example_sim_granular_collision_sdf.py +13 -13
  20. examples/example_sim_neo_hookean.py +3 -3
  21. examples/example_sim_particle_chain.py +2 -2
  22. examples/example_sim_quadruped.py +8 -5
  23. examples/example_sim_rigid_chain.py +8 -5
  24. examples/example_sim_rigid_contact.py +13 -10
  25. examples/example_sim_rigid_fem.py +2 -2
  26. examples/example_sim_rigid_gyroscopic.py +2 -2
  27. examples/example_sim_rigid_kinematics.py +1 -1
  28. examples/example_sim_trajopt.py +3 -2
  29. examples/fem/example_apic_fluid.py +5 -7
  30. examples/fem/example_diffusion_mgpu.py +18 -16
  31. warp/__init__.py +3 -2
  32. warp/bin/libwarp-clang.dylib +0 -0
  33. warp/bin/libwarp.dylib +0 -0
  34. warp/build_dll.py +29 -9
  35. warp/builtins.py +206 -7
  36. warp/codegen.py +58 -38
  37. warp/config.py +3 -1
  38. warp/context.py +234 -128
  39. warp/fem/__init__.py +2 -2
  40. warp/fem/cache.py +2 -1
  41. warp/fem/field/nodal_field.py +18 -17
  42. warp/fem/geometry/hexmesh.py +11 -6
  43. warp/fem/geometry/quadmesh_2d.py +16 -12
  44. warp/fem/geometry/tetmesh.py +19 -8
  45. warp/fem/geometry/trimesh_2d.py +18 -7
  46. warp/fem/integrate.py +341 -196
  47. warp/fem/quadrature/__init__.py +1 -1
  48. warp/fem/quadrature/pic_quadrature.py +138 -53
  49. warp/fem/quadrature/quadrature.py +81 -9
  50. warp/fem/space/__init__.py +1 -1
  51. warp/fem/space/basis_space.py +169 -51
  52. warp/fem/space/grid_2d_function_space.py +2 -2
  53. warp/fem/space/grid_3d_function_space.py +2 -2
  54. warp/fem/space/hexmesh_function_space.py +2 -2
  55. warp/fem/space/partition.py +9 -6
  56. warp/fem/space/quadmesh_2d_function_space.py +2 -2
  57. warp/fem/space/shape/cube_shape_function.py +27 -15
  58. warp/fem/space/shape/square_shape_function.py +29 -18
  59. warp/fem/space/tetmesh_function_space.py +2 -2
  60. warp/fem/space/topology.py +10 -0
  61. warp/fem/space/trimesh_2d_function_space.py +2 -2
  62. warp/fem/utils.py +10 -5
  63. warp/native/array.h +49 -8
  64. warp/native/builtin.h +31 -14
  65. warp/native/cuda_util.cpp +8 -3
  66. warp/native/cuda_util.h +1 -0
  67. warp/native/exports.h +1177 -1108
  68. warp/native/intersect.h +4 -4
  69. warp/native/intersect_adj.h +8 -8
  70. warp/native/mat.h +65 -6
  71. warp/native/mesh.h +126 -5
  72. warp/native/quat.h +28 -4
  73. warp/native/vec.h +76 -14
  74. warp/native/warp.cu +1 -6
  75. warp/render/render_opengl.py +261 -109
  76. warp/sim/import_mjcf.py +13 -7
  77. warp/sim/import_urdf.py +14 -14
  78. warp/sim/inertia.py +17 -18
  79. warp/sim/model.py +67 -67
  80. warp/sim/render.py +1 -1
  81. warp/sparse.py +6 -6
  82. warp/stubs.py +19 -81
  83. warp/tape.py +1 -1
  84. warp/tests/__main__.py +3 -6
  85. warp/tests/{test_class_kernel.py → aux_test_class_kernel.py} +9 -1
  86. warp/tests/aux_test_conditional_unequal_types_kernels.py +21 -0
  87. warp/tests/{test_dependent.py → aux_test_dependent.py} +2 -2
  88. warp/tests/{test_reference.py → aux_test_reference.py} +1 -1
  89. warp/tests/aux_test_unresolved_func.py +14 -0
  90. warp/tests/aux_test_unresolved_symbol.py +14 -0
  91. warp/tests/{test_kinematics.py → disabled_kinematics.py} +10 -12
  92. warp/tests/run_coverage_serial.py +31 -0
  93. warp/tests/test_adam.py +102 -106
  94. warp/tests/test_arithmetic.py +39 -40
  95. warp/tests/test_array.py +46 -48
  96. warp/tests/test_array_reduce.py +25 -19
  97. warp/tests/test_atomic.py +62 -26
  98. warp/tests/test_bool.py +16 -11
  99. warp/tests/test_builtins_resolution.py +1292 -0
  100. warp/tests/test_bvh.py +9 -12
  101. warp/tests/test_closest_point_edge_edge.py +53 -57
  102. warp/tests/test_codegen.py +164 -134
  103. warp/tests/test_compile_consts.py +13 -19
  104. warp/tests/test_conditional.py +30 -32
  105. warp/tests/test_copy.py +9 -12
  106. warp/tests/test_ctypes.py +90 -98
  107. warp/tests/test_dense.py +20 -14
  108. warp/tests/test_devices.py +34 -35
  109. warp/tests/test_dlpack.py +74 -75
  110. warp/tests/test_examples.py +215 -97
  111. warp/tests/test_fabricarray.py +15 -21
  112. warp/tests/test_fast_math.py +14 -11
  113. warp/tests/test_fem.py +280 -97
  114. warp/tests/test_fp16.py +19 -15
  115. warp/tests/test_func.py +177 -194
  116. warp/tests/test_generics.py +71 -77
  117. warp/tests/test_grad.py +83 -32
  118. warp/tests/test_grad_customs.py +7 -9
  119. warp/tests/test_hash_grid.py +6 -10
  120. warp/tests/test_import.py +9 -23
  121. warp/tests/test_indexedarray.py +19 -21
  122. warp/tests/test_intersect.py +15 -9
  123. warp/tests/test_large.py +17 -19
  124. warp/tests/test_launch.py +14 -17
  125. warp/tests/test_lerp.py +63 -63
  126. warp/tests/test_lvalue.py +84 -35
  127. warp/tests/test_marching_cubes.py +9 -13
  128. warp/tests/test_mat.py +388 -3004
  129. warp/tests/test_mat_lite.py +9 -12
  130. warp/tests/test_mat_scalar_ops.py +2889 -0
  131. warp/tests/test_math.py +10 -11
  132. warp/tests/test_matmul.py +104 -100
  133. warp/tests/test_matmul_lite.py +72 -98
  134. warp/tests/test_mesh.py +35 -32
  135. warp/tests/test_mesh_query_aabb.py +18 -25
  136. warp/tests/test_mesh_query_point.py +39 -23
  137. warp/tests/test_mesh_query_ray.py +9 -21
  138. warp/tests/test_mlp.py +8 -9
  139. warp/tests/test_model.py +89 -93
  140. warp/tests/test_modules_lite.py +15 -25
  141. warp/tests/test_multigpu.py +87 -114
  142. warp/tests/test_noise.py +10 -12
  143. warp/tests/test_operators.py +14 -21
  144. warp/tests/test_options.py +10 -11
  145. warp/tests/test_pinned.py +16 -18
  146. warp/tests/test_print.py +16 -20
  147. warp/tests/test_quat.py +121 -88
  148. warp/tests/test_rand.py +12 -13
  149. warp/tests/test_reload.py +27 -32
  150. warp/tests/test_rounding.py +7 -10
  151. warp/tests/test_runlength_encode.py +105 -106
  152. warp/tests/test_smoothstep.py +8 -9
  153. warp/tests/test_snippet.py +13 -22
  154. warp/tests/test_sparse.py +30 -29
  155. warp/tests/test_spatial.py +179 -174
  156. warp/tests/test_streams.py +100 -107
  157. warp/tests/test_struct.py +98 -67
  158. warp/tests/test_tape.py +11 -17
  159. warp/tests/test_torch.py +89 -86
  160. warp/tests/test_transient_module.py +9 -12
  161. warp/tests/test_types.py +328 -50
  162. warp/tests/test_utils.py +217 -218
  163. warp/tests/test_vec.py +133 -2133
  164. warp/tests/test_vec_lite.py +8 -11
  165. warp/tests/test_vec_scalar_ops.py +2099 -0
  166. warp/tests/test_volume.py +391 -382
  167. warp/tests/test_volume_write.py +122 -135
  168. warp/tests/unittest_serial.py +35 -0
  169. warp/tests/unittest_suites.py +291 -0
  170. warp/tests/{test_base.py → unittest_utils.py} +138 -25
  171. warp/tests/{test_misc.py → unused_test_misc.py} +13 -5
  172. warp/tests/{test_debug.py → walkthough_debug.py} +2 -15
  173. warp/thirdparty/unittest_parallel.py +257 -54
  174. warp/types.py +119 -98
  175. warp/utils.py +14 -0
  176. {warp_lang-1.0.0b5.dist-info → warp_lang-1.0.0b6.dist-info}/METADATA +2 -1
  177. {warp_lang-1.0.0b5.dist-info → warp_lang-1.0.0b6.dist-info}/RECORD +183 -179
  178. {warp_lang-1.0.0b5.dist-info → warp_lang-1.0.0b6.dist-info}/WHEEL +1 -1
  179. warp/tests/test_all.py +0 -239
  180. warp/tests/test_conditional_unequal_types_kernels.py +0 -14
  181. warp/tests/test_coverage.py +0 -38
  182. warp/tests/test_unresolved_func.py +0 -7
  183. warp/tests/test_unresolved_symbol.py +0 -7
  184. /warp/tests/{test_compile_consts_dummy.py → aux_test_compile_consts_dummy.py} +0 -0
  185. /warp/tests/{test_reference_reference.py → aux_test_reference_reference.py} +0 -0
  186. /warp/tests/{test_square.py → aux_test_square.py} +0 -0
  187. {warp_lang-1.0.0b5.dist-info → warp_lang-1.0.0b6.dist-info}/LICENSE.md +0 -0
  188. {warp_lang-1.0.0b5.dist-info → warp_lang-1.0.0b6.dist-info}/top_level.txt +0 -0
warp/tests/test_mat_lite.py CHANGED
@@ -1,4 +1,4 @@
1
- # Copyright (c) 2022 NVIDIA CORPORATION. All rights reserved.
1
+ # Copyright (c) 2023 NVIDIA CORPORATION. All rights reserved.
2
2
  # NVIDIA CORPORATION and its licensors retain all intellectual property
3
3
  # and proprietary rights in and to this software, related documentation
4
4
  # and any modifications thereto. Any use, reproduction, disclosure or
@@ -7,9 +7,8 @@
7
7
 
8
8
  import unittest
9
9
 
10
- import numpy as np
11
10
  import warp as wp
12
- from warp.tests.test_base import *
11
+ from warp.tests.unittest_utils import *
13
12
 
14
13
  wp.init()
15
14
 
@@ -99,20 +98,18 @@ def test_matrix_mutation(expected: wp.types.matrix(shape=(10, 3), dtype=float)):
99
98
  wp.expect_eq(m, expected)
100
99
 
101
100
 
102
- def register(parent):
103
- devices = get_test_devices()
101
+ devices = get_test_devices()
104
102
 
105
- class TestMat(parent):
106
- pass
107
103
 
108
- add_kernel_test(TestMat, test_matrix_constructor_value_func, dim=1, devices=devices)
109
- add_kernel_test(TestMat, test_constructors_explicit_precision, dim=1, devices=devices)
110
- add_kernel_test(TestMat, test_constructors_default_precision, dim=1, devices=devices)
104
+ class TestMatLite(unittest.TestCase):
105
+ pass
111
106
 
112
- return TestMat
107
+
108
+ add_kernel_test(TestMatLite, test_matrix_constructor_value_func, dim=1, devices=devices)
109
+ add_kernel_test(TestMatLite, test_constructors_explicit_precision, dim=1, devices=devices)
110
+ add_kernel_test(TestMatLite, test_constructors_default_precision, dim=1, devices=devices)
113
111
 
114
112
 
115
113
  if __name__ == "__main__":
116
114
  wp.build.clear_kernel_cache()
117
- _ = register(unittest.TestCase)
118
115
  unittest.main(verbosity=2, failfast=True)