warp-lang 1.0.0b2__py3-none-manylinux2014_x86_64.whl → 1.0.0b6__py3-none-manylinux2014_x86_64.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.
- docs/conf.py +17 -5
- examples/env/env_ant.py +1 -1
- examples/env/env_cartpole.py +1 -1
- examples/env/env_humanoid.py +1 -1
- examples/env/env_usd.py +4 -1
- examples/env/environment.py +8 -9
- examples/example_dem.py +34 -33
- examples/example_diffray.py +364 -337
- examples/example_fluid.py +32 -23
- examples/example_jacobian_ik.py +97 -93
- examples/example_marching_cubes.py +6 -16
- examples/example_mesh.py +6 -16
- examples/example_mesh_intersect.py +16 -14
- examples/example_nvdb.py +14 -16
- examples/example_raycast.py +14 -13
- examples/example_raymarch.py +16 -23
- examples/example_render_opengl.py +19 -10
- examples/example_sim_cartpole.py +82 -78
- examples/example_sim_cloth.py +45 -48
- examples/example_sim_fk_grad.py +51 -44
- examples/example_sim_fk_grad_torch.py +47 -40
- examples/example_sim_grad_bounce.py +108 -133
- examples/example_sim_grad_cloth.py +99 -113
- examples/example_sim_granular.py +5 -6
- examples/{example_sim_sdf_shape.py → example_sim_granular_collision_sdf.py} +37 -26
- examples/example_sim_neo_hookean.py +51 -55
- examples/example_sim_particle_chain.py +4 -4
- examples/example_sim_quadruped.py +126 -81
- examples/example_sim_rigid_chain.py +54 -61
- examples/example_sim_rigid_contact.py +66 -70
- examples/example_sim_rigid_fem.py +3 -3
- examples/example_sim_rigid_force.py +1 -1
- examples/example_sim_rigid_gyroscopic.py +3 -4
- examples/example_sim_rigid_kinematics.py +28 -39
- examples/example_sim_trajopt.py +112 -110
- examples/example_sph.py +9 -8
- examples/example_wave.py +7 -7
- examples/fem/bsr_utils.py +30 -17
- examples/fem/example_apic_fluid.py +85 -69
- examples/fem/example_convection_diffusion.py +97 -93
- examples/fem/example_convection_diffusion_dg.py +142 -149
- examples/fem/example_convection_diffusion_dg0.py +141 -136
- examples/fem/example_deformed_geometry.py +146 -0
- examples/fem/example_diffusion.py +115 -84
- examples/fem/example_diffusion_3d.py +116 -86
- examples/fem/example_diffusion_mgpu.py +102 -79
- examples/fem/example_mixed_elasticity.py +139 -100
- examples/fem/example_navier_stokes.py +175 -162
- examples/fem/example_stokes.py +143 -111
- examples/fem/example_stokes_transfer.py +186 -157
- examples/fem/mesh_utils.py +59 -97
- examples/fem/plot_utils.py +138 -17
- tools/ci/publishing/build_nodes_info.py +54 -0
- warp/__init__.py +4 -3
- warp/__init__.pyi +1 -0
- warp/bin/warp-clang.so +0 -0
- warp/bin/warp.so +0 -0
- warp/build.py +5 -3
- warp/build_dll.py +29 -9
- warp/builtins.py +836 -492
- warp/codegen.py +864 -553
- warp/config.py +3 -1
- warp/context.py +389 -172
- warp/fem/__init__.py +24 -6
- warp/fem/cache.py +318 -25
- warp/fem/dirichlet.py +7 -3
- warp/fem/domain.py +14 -0
- warp/fem/field/__init__.py +30 -38
- warp/fem/field/field.py +149 -0
- warp/fem/field/nodal_field.py +244 -138
- warp/fem/field/restriction.py +8 -6
- warp/fem/field/test.py +127 -59
- warp/fem/field/trial.py +117 -60
- warp/fem/geometry/__init__.py +5 -1
- warp/fem/geometry/deformed_geometry.py +271 -0
- warp/fem/geometry/element.py +24 -1
- warp/fem/geometry/geometry.py +86 -14
- warp/fem/geometry/grid_2d.py +112 -54
- warp/fem/geometry/grid_3d.py +134 -65
- warp/fem/geometry/hexmesh.py +953 -0
- warp/fem/geometry/partition.py +85 -33
- warp/fem/geometry/quadmesh_2d.py +532 -0
- warp/fem/geometry/tetmesh.py +451 -115
- warp/fem/geometry/trimesh_2d.py +197 -92
- warp/fem/integrate.py +534 -268
- warp/fem/operator.py +58 -31
- warp/fem/polynomial.py +11 -0
- warp/fem/quadrature/__init__.py +1 -1
- warp/fem/quadrature/pic_quadrature.py +150 -58
- warp/fem/quadrature/quadrature.py +209 -57
- warp/fem/space/__init__.py +230 -53
- warp/fem/space/basis_space.py +489 -0
- warp/fem/space/collocated_function_space.py +105 -0
- warp/fem/space/dof_mapper.py +49 -2
- warp/fem/space/function_space.py +90 -39
- warp/fem/space/grid_2d_function_space.py +149 -496
- warp/fem/space/grid_3d_function_space.py +173 -538
- warp/fem/space/hexmesh_function_space.py +352 -0
- warp/fem/space/partition.py +129 -76
- warp/fem/space/quadmesh_2d_function_space.py +369 -0
- warp/fem/space/restriction.py +46 -34
- warp/fem/space/shape/__init__.py +15 -0
- warp/fem/space/shape/cube_shape_function.py +738 -0
- warp/fem/space/shape/shape_function.py +103 -0
- warp/fem/space/shape/square_shape_function.py +611 -0
- warp/fem/space/shape/tet_shape_function.py +567 -0
- warp/fem/space/shape/triangle_shape_function.py +429 -0
- warp/fem/space/tetmesh_function_space.py +132 -1039
- warp/fem/space/topology.py +295 -0
- warp/fem/space/trimesh_2d_function_space.py +104 -742
- warp/fem/types.py +13 -11
- warp/fem/utils.py +335 -60
- warp/native/array.h +120 -34
- warp/native/builtin.h +101 -72
- warp/native/bvh.cpp +73 -325
- warp/native/bvh.cu +406 -23
- warp/native/bvh.h +22 -40
- warp/native/clang/clang.cpp +1 -0
- warp/native/crt.h +2 -0
- warp/native/cuda_util.cpp +8 -3
- warp/native/cuda_util.h +1 -0
- warp/native/exports.h +1522 -1243
- warp/native/intersect.h +19 -4
- warp/native/intersect_adj.h +8 -8
- warp/native/mat.h +76 -17
- warp/native/mesh.cpp +33 -108
- warp/native/mesh.cu +114 -18
- warp/native/mesh.h +395 -40
- warp/native/noise.h +272 -329
- warp/native/quat.h +51 -8
- warp/native/rand.h +44 -34
- warp/native/reduce.cpp +1 -1
- warp/native/sparse.cpp +4 -4
- warp/native/sparse.cu +163 -155
- warp/native/spatial.h +2 -2
- warp/native/temp_buffer.h +18 -14
- warp/native/vec.h +103 -21
- warp/native/warp.cpp +2 -1
- warp/native/warp.cu +28 -3
- warp/native/warp.h +4 -3
- warp/render/render_opengl.py +261 -109
- warp/sim/__init__.py +1 -2
- warp/sim/articulation.py +385 -185
- warp/sim/import_mjcf.py +59 -48
- warp/sim/import_urdf.py +15 -15
- warp/sim/import_usd.py +174 -102
- warp/sim/inertia.py +17 -18
- warp/sim/integrator_xpbd.py +4 -3
- warp/sim/model.py +330 -250
- warp/sim/render.py +1 -1
- warp/sparse.py +625 -152
- warp/stubs.py +341 -309
- warp/tape.py +9 -6
- warp/tests/__main__.py +3 -6
- warp/tests/assets/curlnoise_golden.npy +0 -0
- warp/tests/assets/pnoise_golden.npy +0 -0
- warp/tests/{test_class_kernel.py → aux_test_class_kernel.py} +9 -1
- warp/tests/aux_test_conditional_unequal_types_kernels.py +21 -0
- warp/tests/{test_dependent.py → aux_test_dependent.py} +2 -2
- warp/tests/{test_reference.py → aux_test_reference.py} +1 -1
- warp/tests/aux_test_unresolved_func.py +14 -0
- warp/tests/aux_test_unresolved_symbol.py +14 -0
- warp/tests/disabled_kinematics.py +239 -0
- warp/tests/run_coverage_serial.py +31 -0
- warp/tests/test_adam.py +103 -106
- warp/tests/test_arithmetic.py +94 -74
- warp/tests/test_array.py +82 -101
- warp/tests/test_array_reduce.py +57 -23
- warp/tests/test_atomic.py +64 -28
- warp/tests/test_bool.py +22 -12
- warp/tests/test_builtins_resolution.py +1292 -0
- warp/tests/test_bvh.py +18 -18
- warp/tests/test_closest_point_edge_edge.py +54 -57
- warp/tests/test_codegen.py +165 -134
- warp/tests/test_compile_consts.py +28 -20
- warp/tests/test_conditional.py +108 -24
- warp/tests/test_copy.py +10 -12
- warp/tests/test_ctypes.py +112 -88
- warp/tests/test_dense.py +21 -14
- warp/tests/test_devices.py +98 -0
- warp/tests/test_dlpack.py +75 -75
- warp/tests/test_examples.py +237 -0
- warp/tests/test_fabricarray.py +22 -24
- warp/tests/test_fast_math.py +15 -11
- warp/tests/test_fem.py +1034 -124
- warp/tests/test_fp16.py +23 -16
- warp/tests/test_func.py +187 -86
- warp/tests/test_generics.py +194 -49
- warp/tests/test_grad.py +123 -181
- warp/tests/test_grad_customs.py +176 -0
- warp/tests/test_hash_grid.py +35 -34
- warp/tests/test_import.py +10 -23
- warp/tests/test_indexedarray.py +24 -25
- warp/tests/test_intersect.py +18 -9
- warp/tests/test_large.py +141 -0
- warp/tests/test_launch.py +14 -41
- warp/tests/test_lerp.py +64 -65
- warp/tests/test_lvalue.py +493 -0
- warp/tests/test_marching_cubes.py +12 -13
- warp/tests/test_mat.py +517 -2898
- warp/tests/test_mat_lite.py +115 -0
- warp/tests/test_mat_scalar_ops.py +2889 -0
- warp/tests/test_math.py +103 -9
- warp/tests/test_matmul.py +304 -69
- warp/tests/test_matmul_lite.py +410 -0
- warp/tests/test_mesh.py +60 -22
- warp/tests/test_mesh_query_aabb.py +21 -25
- warp/tests/test_mesh_query_point.py +111 -22
- warp/tests/test_mesh_query_ray.py +12 -24
- warp/tests/test_mlp.py +30 -22
- warp/tests/test_model.py +92 -89
- warp/tests/test_modules_lite.py +39 -0
- warp/tests/test_multigpu.py +88 -114
- warp/tests/test_noise.py +12 -11
- warp/tests/test_operators.py +16 -20
- warp/tests/test_options.py +11 -11
- warp/tests/test_pinned.py +17 -18
- warp/tests/test_print.py +32 -11
- warp/tests/test_quat.py +275 -129
- warp/tests/test_rand.py +18 -16
- warp/tests/test_reload.py +38 -34
- warp/tests/test_rounding.py +50 -43
- warp/tests/test_runlength_encode.py +168 -20
- warp/tests/test_smoothstep.py +9 -11
- warp/tests/test_snippet.py +143 -0
- warp/tests/test_sparse.py +261 -63
- warp/tests/test_spatial.py +276 -243
- warp/tests/test_streams.py +110 -85
- warp/tests/test_struct.py +268 -63
- warp/tests/test_tape.py +39 -21
- warp/tests/test_torch.py +90 -86
- warp/tests/test_transient_module.py +10 -12
- warp/tests/test_types.py +363 -0
- warp/tests/test_utils.py +451 -0
- warp/tests/test_vec.py +354 -2050
- warp/tests/test_vec_lite.py +73 -0
- warp/tests/test_vec_scalar_ops.py +2099 -0
- warp/tests/test_volume.py +418 -376
- warp/tests/test_volume_write.py +124 -134
- warp/tests/unittest_serial.py +35 -0
- warp/tests/unittest_suites.py +291 -0
- warp/tests/unittest_utils.py +342 -0
- warp/tests/{test_misc.py → unused_test_misc.py} +13 -5
- warp/tests/{test_debug.py → walkthough_debug.py} +3 -17
- warp/thirdparty/appdirs.py +36 -45
- warp/thirdparty/unittest_parallel.py +589 -0
- warp/types.py +622 -211
- warp/utils.py +54 -393
- warp_lang-1.0.0b6.dist-info/METADATA +238 -0
- warp_lang-1.0.0b6.dist-info/RECORD +409 -0
- {warp_lang-1.0.0b2.dist-info → warp_lang-1.0.0b6.dist-info}/WHEEL +1 -1
- examples/example_cache_management.py +0 -40
- examples/example_multigpu.py +0 -54
- examples/example_struct.py +0 -65
- examples/fem/example_stokes_transfer_3d.py +0 -210
- warp/fem/field/discrete_field.py +0 -80
- warp/fem/space/nodal_function_space.py +0 -233
- warp/tests/test_all.py +0 -223
- warp/tests/test_array_scan.py +0 -60
- warp/tests/test_base.py +0 -208
- warp/tests/test_unresolved_func.py +0 -7
- warp/tests/test_unresolved_symbol.py +0 -7
- warp_lang-1.0.0b2.dist-info/METADATA +0 -26
- warp_lang-1.0.0b2.dist-info/RECORD +0 -378
- /warp/tests/{test_compile_consts_dummy.py → aux_test_compile_consts_dummy.py} +0 -0
- /warp/tests/{test_reference_reference.py → aux_test_reference_reference.py} +0 -0
- /warp/tests/{test_square.py → aux_test_square.py} +0 -0
- {warp_lang-1.0.0b2.dist-info → warp_lang-1.0.0b6.dist-info}/LICENSE.md +0 -0
- {warp_lang-1.0.0b2.dist-info → warp_lang-1.0.0b6.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,409 @@
|
|
|
1
|
+
docs/conf.py,sha256=FHItR6eNAj6TeuWRo--2UjFrfM_dFTZmeZkMJ12e3L8,4763
|
|
2
|
+
examples/benchmark_api.py,sha256=3W6ETJyumNs3jfzWecDdCcHfiGmFatcG66-EkHwTeiI,3894
|
|
3
|
+
examples/benchmark_launches.py,sha256=dlsGCZDeqp9K3p5DDjLK8_d6qv54jgs1aI9_AKDTrVg,6190
|
|
4
|
+
examples/example_dem.py,sha256=dTNH2sl3XBTK6BQMH9W-vvOP7_Vhv5bePqBTyjJ2coI,7941
|
|
5
|
+
examples/example_diffray.py,sha256=eqWg2OPTARDIsCwfy1YlIEVTE4dklOae3o54HaavFx0,17681
|
|
6
|
+
examples/example_fluid.py,sha256=ZNTCKTsS0KtYim8pCtFcT26mSXSWx7pXTQQiMNW2WNI,7571
|
|
7
|
+
examples/example_jacobian_ik.py,sha256=YQRQ0Pa90Zna_KRk4xnHs4x6GEcniYCwD-k8mM4PI9s,9300
|
|
8
|
+
examples/example_marching_cubes.py,sha256=fNQ3Ag82vGK7Vpnv6izDxOvze-i_mLM_nMYTdn71Gvg,3699
|
|
9
|
+
examples/example_mesh.py,sha256=2CKxwilvD-wViU2-i5rBgLxgmB6ijB__L3usIwHT7NA,4619
|
|
10
|
+
examples/example_mesh_intersect.py,sha256=Y2dB_YIZNah0FqN443MMWHobQGWOqkjWZOrxiA0AIyk,6431
|
|
11
|
+
examples/example_nvdb.py,sha256=7bXmhVRw3vgTTD3x9tVw_SP3L11r70-x00UAigOxT2Q,5329
|
|
12
|
+
examples/example_raycast.py,sha256=rOzQMphHyBIIlE8zGzuTCEjKiIlpfOHCA2hs8bPHNm8,2854
|
|
13
|
+
examples/example_raymarch.py,sha256=qoygHJyd_U_cqhLuBDfs4SDUblYNwGEr989EqKHUdP0,4769
|
|
14
|
+
examples/example_render_opengl.py,sha256=wczc1gQLtML67Bd_QCaq0uCWk6QkpCP2TTkOGtHXNJs,5226
|
|
15
|
+
examples/example_sim_cartpole.py,sha256=2vmDEJrNQAL0T4POlf3uN_u-Q-WQrowPBE-_GGATG9w,6027
|
|
16
|
+
examples/example_sim_cloth.py,sha256=Jdv00Pw1z97T8_PV7huK_pRt6s8ghobqXxjYyiTWuzU,6120
|
|
17
|
+
examples/example_sim_fk_grad.py,sha256=Alp0-xAI6uT6rTuQWIK-uZEJhXw-vmqu92D2XCvVqkY,5095
|
|
18
|
+
examples/example_sim_fk_grad_torch.py,sha256=yG_Yz7c2RvCLPGFW4GxtcQp1LspkIA9RxjSedoLCUqo,6112
|
|
19
|
+
examples/example_sim_grad_bounce.py,sha256=IKkuOeJphfa00gYCw1B4R7FhhjsEQ4ybUMm9_pNefMI,7805
|
|
20
|
+
examples/example_sim_grad_cloth.py,sha256=EvVTY9gglfl_HMC7_FfxTkuYEM34TkPeUG96V00dufs,6584
|
|
21
|
+
examples/example_sim_granular.py,sha256=m7sTR60iwxHqtTJ07RIaORAZvLGeYTxl5Q2hLqOpm98,3142
|
|
22
|
+
examples/example_sim_granular_collision_sdf.py,sha256=vHtQcndtM0C1ym9jg8pqTZZTa9pkGOR-fKqVNSPjPew,5884
|
|
23
|
+
examples/example_sim_neo_hookean.py,sha256=B_e9BrH7i2X-Ukq91a7dC8flqTw4GsZkJKmJDKFGK6Q,4951
|
|
24
|
+
examples/example_sim_particle_chain.py,sha256=vluTrucFrRt5hdoJt8keo0Y70dFFi-TImnaBvYZ7gZg,2968
|
|
25
|
+
examples/example_sim_quadruped.py,sha256=-M35z-F-BP4cd0UBR7NzmKua4muelizTvdwo2NdCOUs,7903
|
|
26
|
+
examples/example_sim_rigid_chain.py,sha256=zoLqhBS3VW9x3FS__sXsoBI8-U_PviYj7z9lHWEAo3s,7294
|
|
27
|
+
examples/example_sim_rigid_contact.py,sha256=3EUM75ggW35-uTmzsdCoHPyrDf3_RzVzlmdgfALwF_w,6182
|
|
28
|
+
examples/example_sim_rigid_fem.py,sha256=pyMwtQRbEwLQmdFxnvDitZiG_ZMR5qWYFAxnHmNu40M,3445
|
|
29
|
+
examples/example_sim_rigid_force.py,sha256=FwP_sMW9o2r25-A2erffmCZ0IQuT6yYwYuEwYnYP32M,2864
|
|
30
|
+
examples/example_sim_rigid_gyroscopic.py,sha256=OOgB8k8rJ0uH4U2A5QsVJP-nSqlh1j19yL1QaUTipgo,2774
|
|
31
|
+
examples/example_sim_rigid_kinematics.py,sha256=_ldenD4Ergy6YYcGop6flfEHBc4yuYFPxZLB-X3S8X0,3875
|
|
32
|
+
examples/example_sim_trajopt.py,sha256=yooaPLZsbbtA3LOMQeVLr5q1VZ9sugB4ruqIvPZv3sQ,7181
|
|
33
|
+
examples/example_sph.py,sha256=qTgn80faHqjZ5fuuQXzJ1XhVPD_zplKeCtr6KHQUXRA,12194
|
|
34
|
+
examples/example_wave.py,sha256=fxqxTwnwnxH3gmAtmHZ-sFVJwCoJTtSC3au3M0gZ_8I,7807
|
|
35
|
+
examples/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
36
|
+
examples/env/env_ant.py,sha256=LXblxVnbh7Eh8yK4l5v1_wzePuHdui7s4J1j_X87ab4,2058
|
|
37
|
+
examples/env/env_cartpole.py,sha256=c59p1SldbTeAAZD6LG3nAy_3YyXBLL71Aw3G2Gv4amM,2059
|
|
38
|
+
examples/env/env_humanoid.py,sha256=8A2jFNUuM9coe2bVsZgYk5U2thVwBvyDGn3bRS-RlvA,2197
|
|
39
|
+
examples/env/env_usd.py,sha256=8wMTft8dI6rPgLiMsOzX6GMMN6aZwdeKBIOBHNTzNT4,3998
|
|
40
|
+
examples/env/environment.py,sha256=qAoZZ_NKPjo6NJ-0tOWme2FsFZenDOeWCOJohP1T_zM,19796
|
|
41
|
+
examples/fem/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
42
|
+
examples/fem/bsr_utils.py,sha256=mrFJ3mBMVbPUAIlAOqSSgfny7mJ6XY3uz9mrPTUYprM,7566
|
|
43
|
+
examples/fem/example_apic_fluid.py,sha256=7rjQjk9U51fozb2tc7Lp3ptdpCcM6DXpIDEn5yoSbJc,13291
|
|
44
|
+
examples/fem/example_convection_diffusion.py,sha256=6K5Dm4X23Pga_hM8ixp8exFWxEtqUUUfKkMbnmBvI3I,5039
|
|
45
|
+
examples/fem/example_convection_diffusion_dg.py,sha256=T993Vr6B1duoNzfekQqoFgeHn53SzEFHxc_4a12ACVQ,6616
|
|
46
|
+
examples/fem/example_convection_diffusion_dg0.py,sha256=HFxts07P_lL_YOxNmPZa1PMD2dvwPixghseRr1RcRDQ,6470
|
|
47
|
+
examples/fem/example_deformed_geometry.py,sha256=-ZWCXTB4XXKtndDK-QadrEJ-KKyE7u5uOoANYlCZ6ow,5102
|
|
48
|
+
examples/fem/example_diffusion.py,sha256=6L-9f6JKF-jqxE167LpW1p7661GtJxQ5xrjDPpU_da0,5902
|
|
49
|
+
examples/fem/example_diffusion_3d.py,sha256=3ke38nmcFet647WABmj2CmfXFxV9Tz6IbMHsusU_wGc,5110
|
|
50
|
+
examples/fem/example_diffusion_mgpu.py,sha256=JXIvZubIugFFXGYbu6r9rGmI8JKTMTEX55HgPKDCtNc,6516
|
|
51
|
+
examples/fem/example_mixed_elasticity.py,sha256=XOii3WPghgdAspkgj1lW-NS4Er9jFHw0TK5W0geMe2I,7177
|
|
52
|
+
examples/fem/example_navier_stokes.py,sha256=rDaYV5BJTF3eZisNSI6381vbhoQNpJtxhSmtxR9iuDE,8283
|
|
53
|
+
examples/fem/example_stokes.py,sha256=YRKLDWrVRpVVnq5UVQ9UizuepwFyLFhoDxkSdle6VjM,6556
|
|
54
|
+
examples/fem/example_stokes_transfer.py,sha256=09DZnUmJdpWV0sr63TYXAKGhXNJEZfyIlwK1sfXYw5k,8585
|
|
55
|
+
examples/fem/mesh_utils.py,sha256=MU_OHnKZ-ZuNZkOLSb4dHDJt1n4FiRuLjcZ8IussQIY,3703
|
|
56
|
+
examples/fem/plot_utils.py,sha256=Q4VikizcscawqTtc5ah1Ncrt2aOhR4hrHh-_pqHFi2E,9133
|
|
57
|
+
tools/ci/publishing/build_nodes_info.py,sha256=P3j3ULk-rn9pYEbVwh2Ojq6rWePAey_koqKueDF_Wd8,1952
|
|
58
|
+
tools/packman/packmanconf.py,sha256=L9QE-O8PPgttJ43QWPqwN5_q3UZGyWKhwLrPaNopdtw,4039
|
|
59
|
+
tools/packman/bootstrap/install_package.py,sha256=sK9yx5mqGNMWhgsQPiIn5qudupRw86kl7DJWUk2VZRQ,5930
|
|
60
|
+
tools/repoman/repoman.py,sha256=wL_EbdiD3Ny8saWhdaNPS42k2XSkGPwrA0ae5FhX_cY,731
|
|
61
|
+
warp/__init__.py,sha256=l5CZa-wLvmFiXEmPcIb5BJ-D6hOq9iKBMAC2oHTdplM,3713
|
|
62
|
+
warp/__init__.pyi,sha256=xRyS7nyFH3H_hKLtF4HCHEHF31pXBjiQrWRyUgw9mpA,22
|
|
63
|
+
warp/build.py,sha256=P5i_8fqiwd8CImhfDbDeZ3I0CnWU9gCtQXwNnloAh9U,4294
|
|
64
|
+
warp/build_dll.py,sha256=ifOzGKrVYStIe8VLQLQS8tjKzTMmgy8gfm_9HFaRD0M,16844
|
|
65
|
+
warp/builtins.py,sha256=Eb3qSVbAx1LwA67kM7w_gRhpCNTR3PXZRJYumDcIBz4,116613
|
|
66
|
+
warp/codegen.py,sha256=5ws31bglnS-3DgjtUGyobJr2qw8_mZ8oh29rLmKJWkM,97863
|
|
67
|
+
warp/config.py,sha256=IZa37rx9TLcTqRkXv3mC6T4a5iQlmKy4Ht3FkZBiJr4,1751
|
|
68
|
+
warp/constants.py,sha256=7kcIJLSp1byeDxqKPHqOSLml_U6DzWe9qTLFMtIIp_E,1138
|
|
69
|
+
warp/context.py,sha256=KSQC55ZWZBCc1eUGcxE7o2IxAY3Kzk-zupEP8sKuRO0,167393
|
|
70
|
+
warp/dlpack.py,sha256=axWuv7Pch-LZeoDrtabhwwq0WRMjNAZZdvjxt_wkjUg,13724
|
|
71
|
+
warp/fabric.py,sha256=eNj7KKOn10t_ukh2LpsWSazotzARDlHZluUxQc5vISY,11360
|
|
72
|
+
warp/jax.py,sha256=uBynTsn3qD7bXaLUw5yVQuS4KRJRPT26wQgAJedTfgE,1557
|
|
73
|
+
warp/sparse.py,sha256=uFudO9WHXKI4C6v1sYnrxqV3sE-FuPRRWAXK6gBWZeo,42647
|
|
74
|
+
warp/stubs.py,sha256=uTNypn3VrPrWYd7Id9NojdlzauwWHg1EobXDpzz6HqE,59202
|
|
75
|
+
warp/tape.py,sha256=HzU7F-7Ik3PY1rPYQod82u1sB4Ina0_xi-D88Klmx0Q,7689
|
|
76
|
+
warp/torch.py,sha256=fyQVQVkdqYLbS9ytlR7Ewb6A8j1g7YDIHPi2daebMyw,9870
|
|
77
|
+
warp/types.py,sha256=_4BdOuBS6n3QGxu9AQt0oClqChIos_4SKbKwlSKq-l0,143540
|
|
78
|
+
warp/utils.py,sha256=vwapm-IzUOPndHLl5LHj4yJkc7-v-T7O7yfb4gCCszA,23447
|
|
79
|
+
warp/bin/warp-clang.so,sha256=VvrJTNsL-heShOboAPuiiFDwPHzgBbLLorhnmm7nsmA,59972664
|
|
80
|
+
warp/bin/warp.so,sha256=YF40b3iHyCkNY1r9kOihqWfV3NTYetgIPqYueQeL_gU,85042696
|
|
81
|
+
warp/fem/__init__.py,sha256=Ei9gXbYd83K-4dO5IzYXTE3Vs186gw0e25YCpAwYgck,1481
|
|
82
|
+
warp/fem/cache.py,sha256=7Ken7yARYk8oHjOh03qOBI_mc_JhwE1s2JC1Kt9adcU,13349
|
|
83
|
+
warp/fem/dirichlet.py,sha256=SFvCq8fcdLxqIfZkR9uxQQniNF3B2rkVazNvvpm-Lls,6273
|
|
84
|
+
warp/fem/domain.py,sha256=jQFduQ10peDUh2gxONozABjmpgWR_Hvla5pzQjUT_z0,8630
|
|
85
|
+
warp/fem/integrate.py,sha256=TfG9p7PIX6Nugx2ixLK0uf-SW2VRAZQkJTusDn4RXNs,58631
|
|
86
|
+
warp/fem/operator.py,sha256=5AEF3HGbFePhWmVUxcwRkYuOf1n75F0FhZ4wJl9Qsc0,6399
|
|
87
|
+
warp/fem/polynomial.py,sha256=tzyTj4KOTD1G6wQScI9Hs-LXUetN2hT44pp0ZPqP-ws,6781
|
|
88
|
+
warp/fem/types.py,sha256=1PaM7122f7oPe_FgYvpqeRI3Qe3feOlyUXn2UY2Uy9E,2243
|
|
89
|
+
warp/fem/utils.py,sha256=dZf9gIuu5Z3qnzrZY30_ECZccz71aZYyR_AQMtauN1A,16123
|
|
90
|
+
warp/fem/field/__init__.py,sha256=dMI7miEjFG26FMXSw6wdw7M7zZasnyBSl8sUbdFJVeM,3693
|
|
91
|
+
warp/fem/field/field.py,sha256=GqxU2xT2xC4y4FjOsGGWwgrbDsbqvHoI2qYHRU7isTg,5190
|
|
92
|
+
warp/fem/field/nodal_field.py,sha256=E67TH1vLO3uBbdgDnYoBjnEL8a7HbiVhQ_lb5wvYE-U,12523
|
|
93
|
+
warp/fem/field/restriction.py,sha256=GoPqafMghPGSI9KJv7pSgpsgc8qa7gvD5kvBkNVSbWc,870
|
|
94
|
+
warp/fem/field/test.py,sha256=nMIUG-PNwrk3nn077ZNhBZbGjRtpFeWP3Bvtu1STeSs,7176
|
|
95
|
+
warp/fem/field/trial.py,sha256=J-CLG1WxdbQggN9olssa-tzyvgWpKHh__Ohg-iH4Dxc,7101
|
|
96
|
+
warp/fem/geometry/__init__.py,sha256=7oxiqxEXga_hvlOK8WkYplKGcQBgWc82b1oq5Uygu8g,453
|
|
97
|
+
warp/fem/geometry/closest_point.py,sha256=IP0QmrpLi3r6VMfxbnjfwAxzGv_3j83YMAtsAFPSYRA,2099
|
|
98
|
+
warp/fem/geometry/deformed_geometry.py,sha256=UTZZvIuqNrClaGwuSJ-FpVhAXwTczkJu9r4vzSz1CAg,11201
|
|
99
|
+
warp/fem/geometry/element.py,sha256=BdWRYsEB5dlZ0ZfcF43O43PNAV997x86k47RGh4-U8A,34790
|
|
100
|
+
warp/fem/geometry/geometry.py,sha256=jKdUMBGY__ws0XeQrQegQjotSbdIH3kvrSgIInxwedM,7474
|
|
101
|
+
warp/fem/geometry/grid_2d.py,sha256=b7g5e17ma25X_MiAiTbsVPnQ7enugCnceGiCBvLOABE,12155
|
|
102
|
+
warp/fem/geometry/grid_3d.py,sha256=1FXCva6xuzai0ip1aNP1ULcijATOeq150WSv7yYjW0c,14935
|
|
103
|
+
warp/fem/geometry/hexmesh.py,sha256=At9cSCHCKOGFv3hGjW0V0oyslT8je6tWmP_2YEnGECU,35781
|
|
104
|
+
warp/fem/geometry/partition.py,sha256=vr-OZJoz0tj92gydl8WOpq55CcRZpwwY1kao7JqLtRA,13083
|
|
105
|
+
warp/fem/geometry/quadmesh_2d.py,sha256=13IQcQkHPvKj7kJvXUzsapOIuCoYtZ1O_O2PcIblEh8,20085
|
|
106
|
+
warp/fem/geometry/tetmesh.py,sha256=91n6eGYXOQoWqsIy8TUrYUyItJoeV2URWAKBuiKh-gg,30646
|
|
107
|
+
warp/fem/geometry/trimesh_2d.py,sha256=wT9nScBRLNZ6dGJxq5mOjijVlYmRz-puP48MlFXEqpU,20686
|
|
108
|
+
warp/fem/quadrature/__init__.py,sha256=JG0TxooqkrUOrpiVMJi-oX0c99Pg_1-HEQdsb997t6s,135
|
|
109
|
+
warp/fem/quadrature/pic_quadrature.py,sha256=kpif2CPNl_JuUET1pqH1O5udJE2Bas8dLJCliQj-_nY,9761
|
|
110
|
+
warp/fem/quadrature/quadrature.py,sha256=_X9X8YJnTpumpBIErR3dHchsA1IYEBethDMUgwvQvOc,10027
|
|
111
|
+
warp/fem/space/__init__.py,sha256=rCjlO823lx7CYxsE2I3G40a-p7lBBqe0a1qwioKvEyk,12398
|
|
112
|
+
warp/fem/space/basis_space.py,sha256=2zC9MNaFWcFoDBp6QJJ_tt3EDCjI58n6P5CyTvg1DzQ,18351
|
|
113
|
+
warp/fem/space/collocated_function_space.py,sha256=W1F2hisVHLmSmJeCSY1-xSGFmE7aMsWqhIMlZzXAFhQ,3888
|
|
114
|
+
warp/fem/space/dof_mapper.py,sha256=-cDQSp548-ptpXe8rh8i8FBs9BhWh7n7Sb-qjakeNu0,7095
|
|
115
|
+
warp/fem/space/function_space.py,sha256=IAm8rp3hHhyVbHNgGDMTgtwRkJZ1eRt0_W4Gebck3NM,4918
|
|
116
|
+
warp/fem/space/grid_2d_function_space.py,sha256=JbgzeWeQTdqGPDPfdN0Si6YBY7TtbGqvlAOtN66a9p4,8992
|
|
117
|
+
warp/fem/space/grid_3d_function_space.py,sha256=3PISOF0bY7geVlgQ4_3rZgxPPUI1r--tNgUMfyj2lDY,10791
|
|
118
|
+
warp/fem/space/hexmesh_function_space.py,sha256=6HNEPnan0giRnpRpDjznHHCLtjMdO7dB3LprSfSvtEk,12441
|
|
119
|
+
warp/fem/space/partition.py,sha256=21XzXIBWfrB4TQjqVGPfD1I9q1D8Kx6CVixhxqsBuCo,13841
|
|
120
|
+
warp/fem/space/quadmesh_2d_function_space.py,sha256=KURxqkC5ob6TwqWhOT5PmSrVZPs7iJb9VS1CwBJMbCo,13461
|
|
121
|
+
warp/fem/space/restriction.py,sha256=bHJAakDcUlTG35cJ188hyDQNDVff2x0DhypVKz5Oep4,6630
|
|
122
|
+
warp/fem/space/tetmesh_function_space.py,sha256=Yih_YdkPIIT7-gjBPpfVxSXazxcviERS85kEwVlFLRA,10718
|
|
123
|
+
warp/fem/space/topology.py,sha256=yR56wQ_76cc6r4FoyPiX1JM5kLzw-tiCq2J1Zunkh7Q,11158
|
|
124
|
+
warp/fem/space/trimesh_2d_function_space.py,sha256=ayp3L6O_TalY9HN5EmCACv223wrbCZvuPHtMg6539tg,7915
|
|
125
|
+
warp/fem/space/shape/__init__.py,sha256=7ioR6AZ54_cFqkDZwLE-c1j9IBp8HGsDyD4e9fIgol8,640
|
|
126
|
+
warp/fem/space/shape/cube_shape_function.py,sha256=1rhoyNTovguw6mhrni8EH5q0i6SMJqs-QZ4c7KNiSr4,27547
|
|
127
|
+
warp/fem/space/shape/shape_function.py,sha256=bXmbfV97XHerfWf-IQudl0-FVLQlFjYSb8je3A72oZw,3391
|
|
128
|
+
warp/fem/space/shape/square_shape_function.py,sha256=8wWn7y-zmM4xY5fvKVnyYF6HrVtH74A-MpU30Nn9ayg,22825
|
|
129
|
+
warp/fem/space/shape/tet_shape_function.py,sha256=obtxrl-6AWpFEgUClbiN6IJzYABKXARCcDmwmLqrX7Y,20979
|
|
130
|
+
warp/fem/space/shape/triangle_shape_function.py,sha256=VBjGqywBanuBKmT1I1FDzyJmIqHq3ritGEahv6fjxzs,15314
|
|
131
|
+
warp/native/array.h,sha256=_IhcwoseEBsrBE8mrSCvOx9lozMvPIJFfe5l7bRhWuY,37009
|
|
132
|
+
warp/native/builtin.h,sha256=iA-RE2TipD4-ElhI5PI1jL9r7HlwW3CKin_suWIDQb0,48874
|
|
133
|
+
warp/native/bvh.cpp,sha256=Pl2oR9rFPFXNqVXjpcqwIGv3zsedpMrwW2wvl5EUUHQ,10719
|
|
134
|
+
warp/native/bvh.cu,sha256=XrJj54Rexp2W0htMtXYXuQgVI1U91nV41Kg-L7dDQZo,19538
|
|
135
|
+
warp/native/bvh.h,sha256=t9XBVv_KVyvBQ5TrnjlK_pIOuhSONLTxEN9MgwpQAxE,10293
|
|
136
|
+
warp/native/crt.cpp,sha256=Quh5ndL6jlIxYeE_DTc-6s0UWyBPCf57qJlOHZDvaDs,1051
|
|
137
|
+
warp/native/crt.h,sha256=14ZOVfMgWkKcMAgFPoc7VHCJjPhrqZ0JbDuLXxCEJRQ,10811
|
|
138
|
+
warp/native/cuda_crt.h,sha256=Yj02mjPPmeQaaGyGDNHG0BNYpqZo4tfTmIF_NSdKpuA,69706
|
|
139
|
+
warp/native/cuda_util.cpp,sha256=QKGHi-dmDT0bNNDmmAwFQOUryiTPuKs1E4wno8rmvgY,17993
|
|
140
|
+
warp/native/cuda_util.h,sha256=FO6EITyHSUUeA8wlGKPctGxTOHYwd6FBRISvhYD9-Q4,6997
|
|
141
|
+
warp/native/cutlass_gemm.cpp,sha256=Z0NQs0Spnv5BpiE4DAOrha1w6N2dTFao8SnecWmHHT4,1142
|
|
142
|
+
warp/native/cutlass_gemm.cu,sha256=Z26Uitxcd4_n8WoyUS0IBZmL1R9ZfaY6QrCWxY_ZSxU,21785
|
|
143
|
+
warp/native/exports.h,sha256=pz0Jo8X-IVeBvXUp6_fWVsEhbZ8_b1VHe7PjPVHXjZY,154504
|
|
144
|
+
warp/native/fabric.h,sha256=gOqP4rNqbrfjVaab3OWv7qsI4G6-fZzj_khMDFw8pPs,6075
|
|
145
|
+
warp/native/hashgrid.cpp,sha256=93rtLqFbs8qCirOQajaTq6zE3oLzrsk9nMsSTBY1p1c,8055
|
|
146
|
+
warp/native/hashgrid.cu,sha256=g08ZawgQl7_b-DX2c-8pa5unzZAKabK1KeoUJPTHQ5k,2277
|
|
147
|
+
warp/native/hashgrid.h,sha256=UW8CLGudtw8KNkYkFlC4gW6htU4z-0KHerSneIoeL28,6311
|
|
148
|
+
warp/native/initializer_array.h,sha256=B0t-xFl6TZzTm7idY2Dhux3VcoRgFoDp6GpMHgZDXLQ,1028
|
|
149
|
+
warp/native/intersect.h,sha256=QtLiEQp0d7KOHBHhNO8xaiBA5G_6aH5sxYaXWgdx1GU,35339
|
|
150
|
+
warp/native/intersect_adj.h,sha256=x_y03YFw7fT1hFMApqn5riJRnh1oR0rgZtTCP5gcX_8,11928
|
|
151
|
+
warp/native/intersect_tri.h,sha256=j1pXi-7sge-4R6goHGzaXZ8vZhrIXBvxhK_qSEk70P8,10517
|
|
152
|
+
warp/native/marching.cpp,sha256=iFAhaALm4dWzYI_YD5H0tfU3FMmqvPWNd8vRTkYlMG4,26
|
|
153
|
+
warp/native/marching.cu,sha256=jSqvW9UD3j1O2TEdxrqPQnDxw_NnpqFN-ory81G3RBI,21823
|
|
154
|
+
warp/native/marching.h,sha256=ZwfOKCmc2RTkbmPtvb1TDFCu1ScRzcpglz2_6MRL16g,16
|
|
155
|
+
warp/native/mat.h,sha256=6eOhoVljcPOaLjkzhIhz0F-FWOXD3hsK57y6dVOXleI,48765
|
|
156
|
+
warp/native/matnn.h,sha256=CkAgKF879VrNvUFUDb8qZ25tt0HurTuoTZoOW6UWIDw,9948
|
|
157
|
+
warp/native/mesh.cpp,sha256=rE5wuTzYovPsogcf363cw4chAFcyl0iqcu2dYDFjQgc,6013
|
|
158
|
+
warp/native/mesh.cu,sha256=kTNaE6wn4cjK_dqrDmuLAiv9ICDiNW84Ed3RBsoCe2U,10835
|
|
159
|
+
warp/native/mesh.h,sha256=rG42qce2SsbAQ0EA7wa_0arrBX23V3or7HbgdYOjHPc,59990
|
|
160
|
+
warp/native/noise.h,sha256=9T_3rWydo8o8GnaAC1jGUnJOMPoQBePAJhQy4zMR7Dc,30612
|
|
161
|
+
warp/native/quat.h,sha256=FT7rm3zwy1qywGZqPW5HFCxsfmzNThEJul6_WAqK_5k,40312
|
|
162
|
+
warp/native/rand.h,sha256=C8Q-mIZ93hoAZ8r7mVwrFjMvP9K1EJBQ4nTPTWyhpzA,10203
|
|
163
|
+
warp/native/range.h,sha256=jQnd9HaP0UPC_7XfPdTBhOqwQ9BiIgy33PFOaISz_DQ,2398
|
|
164
|
+
warp/native/reduce.cpp,sha256=xIDRVi-mIyknDYcy39Nsm2D8DWYN9aWlh-21Pz8wf4c,4616
|
|
165
|
+
warp/native/reduce.cu,sha256=6s4Cps4c4G67s_vgcnf4LxNZCPCzY9amFaW1aNW6Wm0,11035
|
|
166
|
+
warp/native/runlength_encode.cpp,sha256=KrUu57VxZxYBQO5OUn2HYB7Ha-fTe4mfgpiPs4Tv9DQ,1386
|
|
167
|
+
warp/native/runlength_encode.cu,sha256=ATdNMPIn8XApli09HI3hGEWfhqIG3yPEebk2kLtSk2w,1390
|
|
168
|
+
warp/native/scan.cpp,sha256=lWZwg2ofQzF4srnr4LOno8iV2GKfSZG1aPXF2aQCwuo,853
|
|
169
|
+
warp/native/scan.cu,sha256=QeZ3mgd0u8_AXWZrnzbVsemmW890HS-EYH4z91AqE7s,1224
|
|
170
|
+
warp/native/scan.h,sha256=DUe5rYKyY-g1hscClEehkWaa9EvcJVgbhlAaADwmNyQ,228
|
|
171
|
+
warp/native/solid_angle.h,sha256=RWJW4W90AdjqHWKzzUldZs0Na0yEDITN5SI9BLrX-o0,16127
|
|
172
|
+
warp/native/sort.cpp,sha256=nyZollu6XOcsu15sdGMw2XTbaR3FlwgXvfy--vlG_N8,2273
|
|
173
|
+
warp/native/sort.cu,sha256=R4-VeqyzFS0500RzrNfogvLGqxgLdI049zl_-WpieJE,2700
|
|
174
|
+
warp/native/sort.h,sha256=sodoBbqubyb3ydYTdt-5nqw0LYQ1ScUMvvqFnlY7JX0,709
|
|
175
|
+
warp/native/sparse.cpp,sha256=STqwdjKHvCmwiVIDJYcpGzB4FnsZlQjgrIiLgqh8SVk,13291
|
|
176
|
+
warp/native/sparse.cu,sha256=Q5c9JOJAfrFSY1-MH_KVYf35ZuskQ1mSN5X6J5snZBk,20867
|
|
177
|
+
warp/native/spatial.h,sha256=DEGl45QflOLz9dRfTqXqC6aTTaMQsWs-8c3fEkCmjqc,20423
|
|
178
|
+
warp/native/svd.h,sha256=qqJqPPooYUC6U4fLdRECHAjfQnEfWESdBqeSgcIbntI,21123
|
|
179
|
+
warp/native/temp_buffer.h,sha256=CzHtMTH4713QvdWDchr9tOHf0f2crguln0njupiGc78,528
|
|
180
|
+
warp/native/vec.h,sha256=F1cCFIVUQQ4h3BUKncIHM1y3JUKsXI6oe6B6RKKnGtU,33365
|
|
181
|
+
warp/native/volume.cpp,sha256=VjzVzJ_T4vWPJK4clNTdqdrrfaEwezRBXNYKalRvlEA,9632
|
|
182
|
+
warp/native/volume.cu,sha256=NHnbJaaIE4xUQXwwSSQbnDQNTYavludxluhTnMicIgM,1387
|
|
183
|
+
warp/native/volume.h,sha256=gtG_a0HNxPe4noqsB9aNCsWe8Df-dETGYeIvaH6wuFA,23939
|
|
184
|
+
warp/native/volume_builder.cu,sha256=zWuNrQa39JnrnrQNv456XPz0y5gJ_4V_ZB7iq5wld3Y,20377
|
|
185
|
+
warp/native/volume_builder.h,sha256=GEITFoi0OJmPAWhWRaca_DQBfkvhiT3MsqFM2XAgb60,623
|
|
186
|
+
warp/native/warp.cpp,sha256=hQ1L13yPv3uTRZRTSVdx-j0o7Z8W8-PrUzLyCLxs1vI,33812
|
|
187
|
+
warp/native/warp.cu,sha256=VW0ojkY43bUGT4HcG5si7Cdg4a4cMUaURzHlLQDBwYM,62815
|
|
188
|
+
warp/native/warp.h,sha256=bISLNk-RZ0Mpq1I2Fn7PIRYJT6snd8YFikaPzisMO0A,14911
|
|
189
|
+
warp/native/clang/clang.cpp,sha256=bsjabZFX6EZDgFJJa02KBIp4gpbpVUw-CatERgrqXEg,17699
|
|
190
|
+
warp/native/cutlass/examples/40_cutlass_py/conv2d.py,sha256=mJKDxR_NTRcifad8wDqiMQbxoyhQSUp8Vs_aQlVVmeI,6937
|
|
191
|
+
warp/native/cutlass/examples/40_cutlass_py/gemm.py,sha256=O2boOGEf7h3eGF_AuY5uprZA_sN5pvCWjSEHzWpVsgk,5829
|
|
192
|
+
warp/native/cutlass/examples/40_cutlass_py/gemm_grouped.py,sha256=DaMN_8u1zJmBIw3JvB2JyeeB30ukR-NolKAk2GmAwKw,5818
|
|
193
|
+
warp/native/cutlass/examples/40_cutlass_py/customizable/conv2d.py,sha256=_SiYe0Rhl2fA-b7uXK7yksYNKP5B6yYLFsN5xbI7nMo,15952
|
|
194
|
+
warp/native/cutlass/examples/40_cutlass_py/customizable/gemm.py,sha256=Vgxa6YklsfiT7qW8Nf5WfACEDGXoje3z7ZRTSLun8vA,20840
|
|
195
|
+
warp/native/cutlass/examples/40_cutlass_py/customizable/gemm_grouped.py,sha256=jgZsMiS2IhzRr4d39FgP-EiuRthGLIMuSxf_dDXZc1Y,14007
|
|
196
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_all_code.py,sha256=5YnYb3SUB4mVMjzZAdVbZKIY7fZuhgkSlrBZ9GIb5Ew,5667
|
|
197
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_cmake.py,sha256=y5QKfMSovy1nXm72Yh9cNjPs0AsBblZfMpl1T2M1JZY,4789
|
|
198
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_customized_epilogue.py,sha256=5axkF0N2AoQGz8VXioZDuJXXqAtxlk6DQk4_xKHemNk,4356
|
|
199
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_device.py,sha256=n-mG1qxOuVc0PHFIlA0b6BRIF7rSmG9KXnwLCzfFjtI,20502
|
|
200
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_ir.py,sha256=gWblLP6kUBmcihvtg7attNlLz23neXFDYI_Nhpy8YCE,7623
|
|
201
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_kernel.py,sha256=_O4oQsHaG3ep2_Z5DcGTACVriC4EsNhX2g0V7TutPqw,25104
|
|
202
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_sample.py,sha256=V194wACUDZnV_Zd7zSSMRRfodNBIx0zTDccvQbM5gqM,12068
|
|
203
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_threadblock.py,sha256=5koNIcd_E4xbfzuZOWJyC5yz5NcJKJapTWSLO0ofj0Q,48640
|
|
204
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_turing_and_volta.py,sha256=rchzwfLLAhzloPkIMilMut4pzx5DkBsaCWXGgw3ltqU,21946
|
|
205
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/gen_verify.py,sha256=E1B0TicMQ1wnEGmS_Ex2ptVjM8e5d2jhrB97FsXEpso,3872
|
|
206
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/helper.py,sha256=6AQy8tQYhOrJfaJfkljNtWu1Ct2yXH8eshI9nchubfY,4668
|
|
207
|
+
warp/native/cutlass/examples/44_multi_gemm_ir_and_codegen/ir_gen/replace_fix_impl_header.py,sha256=rgs4r3XMF8clYueYNipqyUiNpB3opLublMd163rWbcY,3179
|
|
208
|
+
warp/native/cutlass/test/unit/gemm/device/simt_sm50.py,sha256=knoqHi9hfx9o0xmMN8dyCXyNDXvyllhTWG3_Xm1A2uQ,17386
|
|
209
|
+
warp/native/cutlass/tools/library/scripts/conv2d_operation.py,sha256=YWLhdjGXLoJB1Ck4e8wdSdIQyOaERcPr0G7WDOO_9o4,18229
|
|
210
|
+
warp/native/cutlass/tools/library/scripts/conv3d_operation.py,sha256=5HTuW477L6uUeVlWtDon9UWbEIRftCVQKNKRENPAB8I,12643
|
|
211
|
+
warp/native/cutlass/tools/library/scripts/gemm_operation.py,sha256=o_c-TYhJxtYTUvrQVghsBUtu3ek6A-VaafkDDf9SgJs,39700
|
|
212
|
+
warp/native/cutlass/tools/library/scripts/generator.py,sha256=zYnMGvX4oqXTmQJPgVM4AS8RR5xn2Srq1-YWSXqbcFo,185889
|
|
213
|
+
warp/native/cutlass/tools/library/scripts/library.py,sha256=m4jquIwbnLd_o4Yrpsg7DMqkyTr8FVeSN6__armsFb4,24859
|
|
214
|
+
warp/native/cutlass/tools/library/scripts/manifest.py,sha256=OAjOzACO9m6s7plTbTkw-GSqKXITyh08QTCob2oDSCg,12743
|
|
215
|
+
warp/native/cutlass/tools/library/scripts/rank_2k_operation.py,sha256=ranCvmmXwEADwobeBWhQFMfay7olWd62N62RBwrVLus,14524
|
|
216
|
+
warp/native/cutlass/tools/library/scripts/rank_k_operation.py,sha256=kD4SFJN3A9NnSg9t7FF-n5tFVyQe0XdIk_ZYN2KWYCk,14096
|
|
217
|
+
warp/native/cutlass/tools/library/scripts/rt.py,sha256=sU1jkiQneowoLX5jHZygfBSRSmr6EAnuJrRidWqb9dg,22416
|
|
218
|
+
warp/native/cutlass/tools/library/scripts/symm_operation.py,sha256=X5pCollHUslibceIRlNAVh7aIlZMjaXpt4mnlf5WUcw,14476
|
|
219
|
+
warp/native/cutlass/tools/library/scripts/trmm_operation.py,sha256=NMsHsLIDJHZPEGREe1UqVx74wuMifk_KpIZPSE8_fmQ,14967
|
|
220
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/setup.py,sha256=ow2zL1qHTt17cvYfJxpRy8MAl1EJxDVMWL5UxQ_ijnI,2545
|
|
221
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/docs/source/conf.py,sha256=lz8fgW3wcBazFEvdk6caTe05i6eN7plFVV3l8kJHpFc,4110
|
|
222
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/profile/conv/conv2d_f16_sm80.py,sha256=aH7aHmlC1FS-2sVES1IwMLCbNB8v4AmnT9HFEBMPvJM,4470
|
|
223
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/profile/gemm/gemm_f32_sm80.py,sha256=IhROR2zzCPRfijoTGLtrXEOpYl8JdRIn81SA-NAOe2w,3747
|
|
224
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/__init__.py,sha256=S9V0maaRmLdHclyqj_YAFSwnTorxw0PCT73r1KJqhJY,1358
|
|
225
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/arguments.py,sha256=kuchtAuf51fD5CTa-eU_rZo9e0AtgvBpuAVachmXTMI,5112
|
|
226
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/c_types.py,sha256=Psmcix6SuGhgNgWZmTFLGoxjJdfz6CW3aQNF90GpVAE,8650
|
|
227
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/compiler.py,sha256=jj2UbhGvVG0a4u8H6PtRmTbb7UuZ1iE45cx-ed66zJk,17334
|
|
228
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/conv2d_operation.py,sha256=fMWTL05hbz5VzSxVVQB4ZEzAyD_N0O3tked4FvgcGK4,25581
|
|
229
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/epilogue.py,sha256=DYOAMPv-A0ZF6t-18zKfm6-NhiqY4d8AhZnEGu1_yyg,40215
|
|
230
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/frontend.py,sha256=OcNb0hYZUZG2Vve5ColgirVA5U8JKlZ_VFE4Qpg6VMs,3589
|
|
231
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/gemm_operation.py,sha256=oDbRrUkqk9yQ0UubwbbP21DtSlrl_zpY3vRWhMz4O1w,51704
|
|
232
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/library.py,sha256=PQjKXofghJPFyD4YTqZYN4_vbwQwVxvVTSkJaW1q-0I,23749
|
|
233
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/memory_manager.py,sha256=pWdqA44y3aDBGeoSM1V4QptsvzYdqRpVl9Eb-YAnTZo,3079
|
|
234
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/operation.py,sha256=0WpEvcHKzmUpmzIaytYCLzHK_q5wIW8pHErKK7q7vls,3934
|
|
235
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/parser.py,sha256=Mg282ZQpTaZ0RL4xCpzGpBWB1-QXW1D9OmXUu6XWm0s,26546
|
|
236
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/reduction_operation.py,sha256=w-LIlZYzUyziF8FJaczTYXfUKHjLZcdr9Zrg5Fc4-qg,15255
|
|
237
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/tensor_ref.py,sha256=GtYSDKvQeE3hwAP34FDIYs_0iAlGSZk9bSvy4QzLOzw,3032
|
|
238
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/type_hint.py,sha256=DGG60i3_BuWnUjN-EspOc9CWfTaVqYgybg3xhn36igA,2007
|
|
239
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/test/__init__.py,sha256=f0jWPEtKTzzLvLDekFAk1CskQsYkEvonbes-fNnTATM,178
|
|
240
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/test/conv2d_testbed.py,sha256=i_v7rAEMdP40ffDKCOjK316VRNulIGxxyWqaMvnZKk0,27219
|
|
241
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/test/gemm_grouped_testbed.py,sha256=2ympp_19qlUPzdiZ-a01i_8mh1lNL1V3Bvv3ou2HU5A,9292
|
|
242
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/test/gemm_testbed.py,sha256=QmYUauUKvR1VJNrG4ZxbQVi1xq5HD7urXSsk2sBmPMQ,22467
|
|
243
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/test/profiler.py,sha256=AYJX_YBuSxTMkNxF6KAIy0UZ41GdYtSun9EYJgvwGpg,3282
|
|
244
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/utils/__init__.py,sha256=HJK2eiB014c0tfXVDRU--A2B19KsCgbhbDhykqCX0L0,48
|
|
245
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/utils/device.py,sha256=QHCwsTqazOxwTfj51kpyAktq9zCadUqIL88m5IBYoYs,3146
|
|
246
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/src/pycutlass/utils/reference_model.py,sha256=-9LdHYoEFHTAHvnJE8cRJo07vQf8GUDVKGxrrLGyqkc,11854
|
|
247
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
248
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.py,sha256=RFeJsdYE9sbVadMPFvrT82HqLPkHbxxyHMknpOpREbY,7962
|
|
249
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=4qMWxgd1vOfgj5RTsgs4Sgut9hLRM1LpJr5Iacx2hEA,7119
|
|
250
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_dgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.py,sha256=PfcPeq97IXTHn-aGlproDEb5o8xUVKv-cE2O4AUr4IQ,3787
|
|
251
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_dgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=tH1FT5salD3QicpQn0x-TlZY1JA-MGtkl1cGGUpj-vE,3783
|
|
252
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_few_channels_f16nhwc_f16nhwc_f16nhwc_tensor_op_f32_sm80.py,sha256=Bf_tEwvSzqyc6BksnzI9Tjh3ZGdSWUil3Bj7SAyMrM0,6504
|
|
253
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_fixed_channels_f16nhwc_f16nhwc_f16nhwc_tensor_op_f32_sm80.py,sha256=9PMlo_ksHOzm37CVNcebbiJts6f6JskOK673KbBhK9A,7493
|
|
254
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.py,sha256=z9cQZg0_wMqdpd9YWwEu-QaOw2XCryxYLlq2lt6hyDc,12337
|
|
255
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=9Irz87iarmoZk44CgECGEqLLOjO-UaYvk-cD3_Wl2UM,2143
|
|
256
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.py,sha256=w4QAY5l82jrcIBbd6mOE1hMu2-JYLzHfIWcw1tZ2IME,3781
|
|
257
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_fprop_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=QE8DtNDE3MCEzoPDvBlWRuFtC4PfRfpxDcrh0mygpHI,4239
|
|
258
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_strided_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=417unadN8FwbOAyApGZyEgR0zOFJtuBK-PrlrNg39wk,10193
|
|
259
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.py,sha256=skPi6ytOqiYZ1pLX5dShNvI5S5CV7Kfay4aRoW6ZtY8,3786
|
|
260
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=lP_tI0kirpTguWPsfvuEFTHi680UfmVlNSYDGjZbXP8,9580
|
|
261
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_wgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.py,sha256=pzbqwuBCpDMcU_uSqIDXHOzfYus0-uEy0CQsykPTpRI,3781
|
|
262
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/conv2d_wgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.py,sha256=FvYALi_Rz3mbqE0NnnI_Y6PdoDH9x3l0pWkqCGF4iDY,4240
|
|
263
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/conv/run_all_tests.py,sha256=ywOq-KAG9tqmS6BBCw3fo_4-xSX1xezAeAfFAZTSGrw,314
|
|
264
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/frontend/test_frontend.py,sha256=u8M45s5a-bjibnjLap5lpqQ9JGeEK9E2A4qpCt0zENo,5727
|
|
265
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
266
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_bf16_sm80.py,sha256=pDRqHzvDnBRgc0K-2kQuE0PamhH6ROXKnGnJiVbDsvU,3421
|
|
267
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_f16_sm80.py,sha256=lSVbIhgQyD8Ef9n-CU8ymvH8kAKABIvHml61MYqbCDc,15900
|
|
268
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_f32_sm80.py,sha256=kX9DHVbewzLApUQrYWq3aGx97bkvuLI3BqeMa5LCQPs,5190
|
|
269
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_f64_sm80.py,sha256=MqUL4YPsx2vOG3hVxk0G0cPs-dXUyu-H9ICa_5FQgyo,3585
|
|
270
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_grouped_sm80.py,sha256=DSNi-3csmkQrGtoxhvHxG7sltWsB0Z31W4lA0eUEE0Q,7094
|
|
271
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/gemm_s8_sm80.py,sha256=6bgSvCBLahgMu3W7cDUhcAgDYKIYghqU7g-joGdZLJ8,7927
|
|
272
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/gemm/run_all_tests.py,sha256=Oh9WnCi_cAhIW95Ic4TIGM4Sx-eCqgZ1BCUWquRSQh4,272
|
|
273
|
+
warp/native/cutlass/tools/library/scripts/pycutlass/test/unit/test_sm80.py,sha256=AYu9gDs0Y8cS1Q8JW5lf788c6j2bLmCID9JQxiKA2ps,19599
|
|
274
|
+
warp/native/nanovdb/NanoVDB.h,sha256=GEqu_o6INERGmevjXetLlCpGw_Ws9q-Dy2ddj__sfok,194604
|
|
275
|
+
warp/native/nanovdb/PNanoVDB.h,sha256=pwjV_boDULb8FTG_NSSWo_HKeD_s5Gw0RGhieVTbEIY,124169
|
|
276
|
+
warp/native/nanovdb/PNanoVDBWrite.h,sha256=hIL4G-ObW_GuuL9DuGX20CEB6U4GY5zk7Sd91RD9prY,18336
|
|
277
|
+
warp/optim/__init__.py,sha256=zbioJc8XHMnfMHg00a9G53UFY3th-dRkou48dpanCQQ,477
|
|
278
|
+
warp/optim/adam.py,sha256=dIiGnuFO-XhMk3egU6PU-QHYvY_pw5W3Qzbbyy6xnKA,4359
|
|
279
|
+
warp/optim/sgd.py,sha256=XnP6tJ5sHc4zplrbHTjo0ynkKLdwQMAJlvRCHf_Sgd8,3119
|
|
280
|
+
warp/render/__init__.py,sha256=VGLibelbTuGcxGhzIiCbGZr6CF-o6z42cr04Pf_WMUk,548
|
|
281
|
+
warp/render/render_opengl.py,sha256=oKRQ5mWPuOMthQLE53_u6aiJj9-4flVxfFa9-seZosg,116744
|
|
282
|
+
warp/render/render_usd.py,sha256=pIcbjBrrgACDsN5g_PZ2TGEAUwZ12Pbiv3pgV82aCKM,25904
|
|
283
|
+
warp/render/utils.py,sha256=Pu-DSAY5jFz7bmX1ZDUJM0TwLr6_lyjfTaW8M-5-mZ4,4446
|
|
284
|
+
warp/sim/__init__.py,sha256=9x3EJ46IeqId6VefBHeonExGf0NHU8WcOZj5XZcRfLY,1716
|
|
285
|
+
warp/sim/articulation.py,sha256=WJSL2-M51i7w1jdCLAr4VgkVKMsTF-11tX5C7G-YmNs,24228
|
|
286
|
+
warp/sim/collide.py,sha256=hTGWd1k0BYkevW0xRHwlAhjYhdqkOHq_5IaNBE_2_vc,57274
|
|
287
|
+
warp/sim/import_mjcf.py,sha256=43z2LfO09ylgAHisFhK8AKzM9PZuimcP0c5gZiAT7PM,20091
|
|
288
|
+
warp/sim/import_snu.py,sha256=lzKknSNbGs_0KaOPH4CctpMsjhXAbijs1zbtflwUKmM,8564
|
|
289
|
+
warp/sim/import_urdf.py,sha256=5kbyTabhMScWpdKsTQxdAjMCWm3YeCOfK2MxmGJbJyo,21089
|
|
290
|
+
warp/sim/import_usd.py,sha256=q8jRdpAcUIqFxL8mnny6p7k628DnJe0O_BZMeo9x-GE,40517
|
|
291
|
+
warp/sim/inertia.py,sha256=m9tagvIdR4uuKPTI4ZYdw-x0S_8PEQMHM7Eq5D_rCvM,9332
|
|
292
|
+
warp/sim/integrator_euler.py,sha256=RCDqit04bCpnJjETkNuH__MrOAavrSzC76nhy4egrgY,61057
|
|
293
|
+
warp/sim/integrator_xpbd.py,sha256=it2JjGkYoHgh5-x5d5t179i-9QCXU2taQyXjrRaXNvg,91871
|
|
294
|
+
warp/sim/model.py,sha256=Z_Y96xtAyGSFCkFvIyOo4-yZbwxsa7r9dU0WsoaEcg4,166392
|
|
295
|
+
warp/sim/optimizer.py,sha256=eGovWgWDCeONGZE_MfqLu-JGXmSACMfKGOfqJ25k2_M,4040
|
|
296
|
+
warp/sim/particles.py,sha256=e_kj6wjjXkj3KhsqepwWPmRc6NoWGqoJZthmnkKr_5A,3174
|
|
297
|
+
warp/sim/render.py,sha256=NbVxLsC0iUfRhzLpXEqEbOQyYBSiRaoHa0-Fip9ClEM,16796
|
|
298
|
+
warp/sim/utils.py,sha256=iDDh3Ug-eazYhzGnfIVtnulyxMZMWQoAQWY3UlH_IIU,5124
|
|
299
|
+
warp/tests/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
|
|
300
|
+
warp/tests/__main__.py,sha256=Dg7dxwx_tD2ORJK4ctN-nmXJ2ilcoba-iLN8fg-tLgE,94
|
|
301
|
+
warp/tests/aux_test_class_kernel.py,sha256=KPjFlrvuVObuGpH5xBf8WwGHP0xg1MM2-Zf0PYmHqHk,1053
|
|
302
|
+
warp/tests/aux_test_compile_consts_dummy.py,sha256=mF4jTs4zQQiycCjbPhTJtkfSj6oCFfodD9sTOKCUyEg,481
|
|
303
|
+
warp/tests/aux_test_conditional_unequal_types_kernels.py,sha256=qVx-16Dg-BDRP0ctHstOC4B_ck6SJcnHfNOKe59vZ2M,667
|
|
304
|
+
warp/tests/aux_test_dependent.py,sha256=_o0Jm-KzbqM0sifQgek6EqSI5yZFm7hVwNF0buRMPrc,743
|
|
305
|
+
warp/tests/aux_test_reference.py,sha256=cXskQIRy9FoH2HhHgfhRSSMfSlbjjbe-uSGx8Gndyeo,216
|
|
306
|
+
warp/tests/aux_test_reference_reference.py,sha256=1A9bHZyBBw3UO0rzAJ6er_utb0c4V6fx5AWR9AezrdQ,141
|
|
307
|
+
warp/tests/aux_test_square.py,sha256=Y4mBorrh7Pt0gHmfvOOvZMbes0NSsHRopca_Wvq7ONw,262
|
|
308
|
+
warp/tests/aux_test_unresolved_func.py,sha256=MqUZqB4xGPsFM4Phj0pPyAazwTx_IKksFhfD6aca-Bc,593
|
|
309
|
+
warp/tests/aux_test_unresolved_symbol.py,sha256=c_ghNYwxm1Ip6RjORSHqs_u8WiQfIyCAl3moczcJGQs,588
|
|
310
|
+
warp/tests/disabled_kinematics.py,sha256=If3joeveVqKIFF6z04BpnD3-TPlYIc4zuwcgrKz3jrI,7851
|
|
311
|
+
warp/tests/run_coverage_serial.py,sha256=14Vej3BouIo3FxAlvOmjAKTN7pjTzVadX3eYNwcvm8Q,1076
|
|
312
|
+
warp/tests/test_adam.py,sha256=cdghsff4zYAtDdXACnY4JoZNKSgiQ5ZSjhBXCDJNuR8,5535
|
|
313
|
+
warp/tests/test_arithmetic.py,sha256=a4gKwIZx7eN_G78JEGYCbqrY-us281FpGrZVwV0jkwc,44951
|
|
314
|
+
warp/tests/test_array.py,sha256=m-RluT3pxOt3Ofg8_N4k_RHkCLcW6hbiEWPkk3Zmlo0,77043
|
|
315
|
+
warp/tests/test_array_reduce.py,sha256=rwR3wI0uWR58p2X7e3e__iEqbtiAqJKl7p4lumz36vw,4835
|
|
316
|
+
warp/tests/test_atomic.py,sha256=huop4ht73RAWxOU8vMeYDEqakiNw4IWhbiFwdol9UTE,5501
|
|
317
|
+
warp/tests/test_bool.py,sha256=7Wa1CBej7VE8b0XcQKRL_jYsYvM8NHo6gdJYRtuPmzA,2527
|
|
318
|
+
warp/tests/test_builtins_resolution.py,sha256=5ClWiJ5tTvfTSfVsPP4P0RE2h-8iE3B1D7PB0ricCGc,65298
|
|
319
|
+
warp/tests/test_bvh.py,sha256=APrFJHk2PAuS3OvPBnxFsCJ6SUJJOLaeqDRqdrUDL2Y,4545
|
|
320
|
+
warp/tests/test_closest_point_edge_edge.py,sha256=0z6FaHz8Z9bVBqpJP1jSsxgH-YRPtHIVsoga1530tdk,7069
|
|
321
|
+
warp/tests/test_codegen.py,sha256=rXDL_LKv_xziXPUutCQ6zeyEj_YlwJ8DfF1Of1Hd7VY,11871
|
|
322
|
+
warp/tests/test_compile_consts.py,sha256=DqhwGYkMp0-NUWjdud03mUO6DinRcNGOuQsBCZdIGOc,2861
|
|
323
|
+
warp/tests/test_conditional.py,sha256=-yVPkiqLzkW3cFMUjO2JkD3YRAI1sPchKv5abnGQKlM,5758
|
|
324
|
+
warp/tests/test_copy.py,sha256=y2yNTQwawANWrm0gYnib18a-KMM9yuJRFKxQKoGG5b4,7261
|
|
325
|
+
warp/tests/test_ctypes.py,sha256=HnWUCfkSfEbCR0xaiDCiUSy03WJRrz1HD79I1KZjxEg,25284
|
|
326
|
+
warp/tests/test_dense.py,sha256=WjBUhrtLyKFCnp3AdBuVeVYZyC3B6AxbUqQhavP5k_I,1913
|
|
327
|
+
warp/tests/test_devices.py,sha256=6n5SAvWPIZTw7KMgQg8-9owUNEPvf9p_9oe6ulDZFO8,3291
|
|
328
|
+
warp/tests/test_dlpack.py,sha256=opQkgRCauWFoRho4qKN4XJQgL6pjEq2FukwHQBzoAvs,13361
|
|
329
|
+
warp/tests/test_examples.py,sha256=9iYJRuhfj9sKMZNsYpJlVNB-S_nxLw_k2lQkltfUHCk,9397
|
|
330
|
+
warp/tests/test_fabricarray.py,sha256=TSKjECqjwltsBkAAnTMhesh-6M47BnXxSphRtwm1WrE,33105
|
|
331
|
+
warp/tests/test_fast_math.py,sha256=vCRlv8XwtQZeZmRmiJlFILKFV64_kNKGUlbsXpJq9c4,1618
|
|
332
|
+
warp/tests/test_fem.py,sha256=xaC08R1KoJb6mKcE_IobZgdEB367c0_P-ohLiL5lD2U,52697
|
|
333
|
+
warp/tests/test_fp16.py,sha256=kIR0SP62vsVLQi1dSezCscn1n3R74l2fxSym6M3Trss,3963
|
|
334
|
+
warp/tests/test_func.py,sha256=c8QVFvetnw0vqi7aMh-vnIs7N99ZPMe3O-WpM5VM9Gg,11071
|
|
335
|
+
warp/tests/test_generics.py,sha256=1o6IrOMlpMkKOXnJz_Koe8aTHKTcW88WyGhe0GjfwJE,19490
|
|
336
|
+
warp/tests/test_grad.py,sha256=5lny5ha8fee6QvvmRlhZpA226knVG_xVk2h1qoxlRpw,21078
|
|
337
|
+
warp/tests/test_grad_customs.py,sha256=3haWrS0e_xw9zAJUmuzMq9TjCcft--y1nDYSvzDFQig,6106
|
|
338
|
+
warp/tests/test_hash_grid.py,sha256=Yw-GYZDAAaS1eTq7ppaS8fht71f4tgyomuPztC3KqAY,4598
|
|
339
|
+
warp/tests/test_import.py,sha256=2T8g-EU1xeEz70pGtKFvFvWGC7znaB2Yn_nvLIgfjTs,1077
|
|
340
|
+
warp/tests/test_indexedarray.py,sha256=5FEsvhOi92RPoq_14H01zmNWK1VcTTWBuuL_xwnoXtg,41945
|
|
341
|
+
warp/tests/test_intersect.py,sha256=CQMPJWi63gu9MkRHvdaWa81yBn2jtX5tqizlChbCeGU,1933
|
|
342
|
+
warp/tests/test_large.py,sha256=GWp1ssEZ9IQ3diSCT3Or1YGCRRLiFs_kFNRaPY8zFx4,4885
|
|
343
|
+
warp/tests/test_launch.py,sha256=grNw0pGfoFfV1_s-1PsMIfatrMMYUn6I-OGQNBmBFns,8642
|
|
344
|
+
warp/tests/test_lerp.py,sha256=vrezacbCx0u5sfMG9jWWxpwUexkh6oyF9Q7c0jeyy2I,7186
|
|
345
|
+
warp/tests/test_lvalue.py,sha256=Ncw4KSCykB4nFdA92VrBtVelGFrBAj7z2YmnFUXnNwo,12070
|
|
346
|
+
warp/tests/test_marching_cubes.py,sha256=VWhYA25wuvRPb7bAYa4VK1kN7O-g4G3qiW-smP450kw,1917
|
|
347
|
+
warp/tests/test_mat.py,sha256=1VYynwQ6sLNqSjZYjB0jdQJ6iFPO2xOiVE7AdiPtex8,67211
|
|
348
|
+
warp/tests/test_mat_lite.py,sha256=MBDLqy7mcqHli9z7qulIgVUghYlW0-vs8VNSsIFeSjs,3909
|
|
349
|
+
warp/tests/test_mat_scalar_ops.py,sha256=CXGE6FUjs2FVbd-ZRdN0M_ng4xh_VHBSUKawXpdK_r0,112805
|
|
350
|
+
warp/tests/test_math.py,sha256=Job9bBiV38FKgYt4vO7FI_OW8mFLjjGkSRqEQSzZDgM,4568
|
|
351
|
+
warp/tests/test_matmul.py,sha256=p0gd5zpQqfGn_oIOu8DrnENmy8CJisjyt51pzEiRNn4,17007
|
|
352
|
+
warp/tests/test_matmul_lite.py,sha256=b2_BDot4pTDjEsNVKejd6carehpa8mkEvfR4AVTss0U,15211
|
|
353
|
+
warp/tests/test_mesh.py,sha256=Wv1U5z76VKJyMWg-10C9t1BR5EHpo_4COsymHugGBW8,8918
|
|
354
|
+
warp/tests/test_mesh_query_aabb.py,sha256=O5adtZLVoC4rSC-kRyUIU1bF34YLZndFtn8DWOGXEB0,6209
|
|
355
|
+
warp/tests/test_mesh_query_point.py,sha256=jSvupZ6d6IWCvRKE7NdXNAjSJksdbBmk3vRQsZuvTHg,21487
|
|
356
|
+
warp/tests/test_mesh_query_ray.py,sha256=EB0oShuz_PXFO-2x-dSbyPYxXQeqNmOy0rvRUSQwz8A,8968
|
|
357
|
+
warp/tests/test_mlp.py,sha256=aOaGsnaHxQY1oZOyfYogSvnYuyeOebxiWhWu8YHW2TI,8299
|
|
358
|
+
warp/tests/test_model.py,sha256=m7zYsTaeXV_dnmZ6ZDLHvLob66QqVlhWTWJZhVYZOJ0,4530
|
|
359
|
+
warp/tests/test_modules_lite.py,sha256=T9pJlkumgu2d-ulLm6GpHJm4Box-3MYY12SC3mDlGok,1209
|
|
360
|
+
warp/tests/test_multigpu.py,sha256=xPtz9bLjZ_IDqCpbIcVwKtS4izmRWbyhtz3775bVX70,5257
|
|
361
|
+
warp/tests/test_noise.py,sha256=jURxSeFnDeTcvdc89PEe14cLb9RvIK9g4IuYwOftLKA,7120
|
|
362
|
+
warp/tests/test_operators.py,sha256=ZPM6dA0C9zdPxCqn6SSSCJ8yy0VkVu6Jj0e7QIBgN7Y,6233
|
|
363
|
+
warp/tests/test_options.py,sha256=AgMOsvaxMRxYOGL7SFUmCP6_WRntvL-2q2iob9h1BYY,2990
|
|
364
|
+
warp/tests/test_pinned.py,sha256=8rRaOtaMVcd7hosrarXYXziUSifhNNOtK6Rzs1UvJPI,2377
|
|
365
|
+
warp/tests/test_print.py,sha256=PCd1SPkG9RHGqxOIoIEiCajASpFeGTxGTcVOPE9KtOs,1559
|
|
366
|
+
warp/tests/test_quat.py,sha256=RDV7ntaVdEmph7aHQzhL5vPRrlt9LLRj9Hqq9dQKKNs,74407
|
|
367
|
+
warp/tests/test_rand.py,sha256=QDNVdr0l92DFHUIej4skt8fB7IwzB40IXAafPkAsjUc,10095
|
|
368
|
+
warp/tests/test_reload.py,sha256=IC8RpzNFx13pSDsZ4VpMIyAKK1Z79OMOCIWVtUwTgnk,5889
|
|
369
|
+
warp/tests/test_rounding.py,sha256=_TyimmIimTyKh5Alxw6utO5Dm9ttmN5DrJDqAbzsnrE,5974
|
|
370
|
+
warp/tests/test_runlength_encode.py,sha256=TVHyYgH1PuiYLJEDk8nCHJLyrR_4NYWI1t7ymDGfWMQ,7155
|
|
371
|
+
warp/tests/test_smoothstep.py,sha256=mL68QQMtV1BsGby4Mywg9tiSx-2vjobTI7spEu-_nUY,4564
|
|
372
|
+
warp/tests/test_snippet.py,sha256=m9DvTKlnehrGbKmO9hsauWeY6Ax9ts9-e8UkH2jHdfA,4220
|
|
373
|
+
warp/tests/test_sparse.py,sha256=Rp6hb6wuBi3TJedy_QZ8znJC_DLTJPEUIN_Agsu17Ew,18333
|
|
374
|
+
warp/tests/test_spatial.py,sha256=YJz7mHnycH8TYq8HTRIETS5jaFaBnOQTCepf6sw7UL4,78561
|
|
375
|
+
warp/tests/test_streams.py,sha256=Ny-n3grNLvHvOqojvHBXL5KukNrUhw1CdECGBzy3dE8,14488
|
|
376
|
+
warp/tests/test_struct.py,sha256=CwsPU4ORD92NpbvVy7dlnhsAimOND2G7FVUvQJWcIFU,16471
|
|
377
|
+
warp/tests/test_tape.py,sha256=NfLEzTkrXVOH1h_vAge-1byEkBZTS-qgTtX-eztUoA8,4717
|
|
378
|
+
warp/tests/test_torch.py,sha256=6eI1Al6S5R9V7bFFx8r39x7q22tMuIiFOlckoKnWwFA,23015
|
|
379
|
+
warp/tests/test_transient_module.py,sha256=z9B78a4B58PRc9b7GtTwSySvQe5IqpmmcZe_STWzCq8,2310
|
|
380
|
+
warp/tests/test_types.py,sha256=1jC6Id1aBN8VGTtmKIad7RDKfalDSj-TmMYUQMeU46M,15461
|
|
381
|
+
warp/tests/test_utils.py,sha256=ZuKitTXKjM5XWMGIi281RMBVez761HV6gTAoPUy4kIk,17619
|
|
382
|
+
warp/tests/test_vec.py,sha256=MQic0V_1hYoEz81p5uo_e0ajhp-2XYGV7S7uRbrusvs,43762
|
|
383
|
+
warp/tests/test_vec_lite.py,sha256=etjM5nRzFxpYCTKtdpr3Qp_HTv4mZRSDQ8JW6ZgxxkA,2427
|
|
384
|
+
warp/tests/test_vec_scalar_ops.py,sha256=z8EDboIPbfA55DoIOhiW4H_VZhXc21AL7xB-aRfP4eM,87005
|
|
385
|
+
warp/tests/test_volume.py,sha256=SYJR8j2Y_Pz4pSN3Ga8KmHUnNr7NJFJhS489Nh-uUqU,25706
|
|
386
|
+
warp/tests/test_volume_write.py,sha256=A7f8M_djIX5jXIr_57l3HCvmIziuqX42S8qyyXgTC4I,9074
|
|
387
|
+
warp/tests/unittest_serial.py,sha256=AJAZPuNwgMMJ_puCvZwgbFiQFN3ZNor95y27UyqkBI0,1145
|
|
388
|
+
warp/tests/unittest_suites.py,sha256=BT4tn1cB72PEZ0L0wGc1WMvGDLYQF27c8LwOxrbZo6Q,10912
|
|
389
|
+
warp/tests/unittest_utils.py,sha256=awEF5o8FIKv_BRKdqOWsIEdZnRqV02OL42Jndb_cSf4,11034
|
|
390
|
+
warp/tests/unused_test_misc.py,sha256=fyfOY3VpF1hSdgu_6kTz2kRSukUBl7LQm-r3-4adJVE,1534
|
|
391
|
+
warp/tests/walkthough_debug.py,sha256=DsotsWkBvkK2hb-iWQRNbAOkK8KkgrdnmyofC5mPy8E,3136
|
|
392
|
+
warp/tests/assets/curlnoise_golden.npy,sha256=ueBhVunJH9If7vdb0fqfxfNaj1fe5pHlKBKHA1TBDRQ,262272
|
|
393
|
+
warp/tests/assets/mlp_golden.npy,sha256=-ynR9GhQN1HaGOYVdkAc-yVBouHN_2nBkW-4FawO8PE,29596
|
|
394
|
+
warp/tests/assets/pnoise_golden.npy,sha256=ZAOYoLXb3C8De8JXo1qwRyUWKjAKmffxegJtzbU4i-k,262272
|
|
395
|
+
warp/tests/assets/spiky.usd,sha256=VYbY7VEDLJxbxMyz8fp6scE4IaT6vvqvwJLnSr7W6h0,1577
|
|
396
|
+
warp/tests/assets/test_grid.nvdb,sha256=lEacSzmPaaMEOnErVZcCcjBzgGjhXbQw_s8MK10h4mQ,15581
|
|
397
|
+
warp/tests/assets/test_int32_grid.nvdb,sha256=WuXLwxwyPso12XVprzy8-JMs0yC-AszvahGmMMEpR9E,15726
|
|
398
|
+
warp/tests/assets/test_vec_grid.nvdb,sha256=c8XefsMg5gbJa3TDfyyD0ImQPkf39yd9CuQkpibXH7U,37620
|
|
399
|
+
warp/tests/assets/torus.nvdb,sha256=7ftuUfUVpR_KkTuBPsmUQUrEFG9BcDxeHK7kvPA3itk,14549
|
|
400
|
+
warp/tests/assets/torus.usda,sha256=rdJoYgz1fyw01sKDHeYIbuTMkq91UL2VlaHCtZ1cWtA,340830
|
|
401
|
+
warp/thirdparty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
402
|
+
warp/thirdparty/appdirs.py,sha256=6msNAwCSJ0eRTVmRVF98Uvd3trshrw5RRsVUM_hPPLE,24852
|
|
403
|
+
warp/thirdparty/dlpack.py,sha256=Upx_8E-DR7TQTHUN0bNbfrilzFubRjxWvafV10ZXvqQ,4416
|
|
404
|
+
warp/thirdparty/unittest_parallel.py,sha256=9CmhJlIiDVjWv5CfXuxm0K_CwezSE_YiCRvk9Fa24J0,22676
|
|
405
|
+
warp_lang-1.0.0b6.dist-info/LICENSE.md,sha256=-oj_azPLVWrkvvTDXjtMSiATqx18KpDmmwXsGmpPSEQ,4110
|
|
406
|
+
warp_lang-1.0.0b6.dist-info/METADATA,sha256=RzZ024mEMol3oZ0v0-GunXydWuTPJV5e-FUqJyiToaU,12412
|
|
407
|
+
warp_lang-1.0.0b6.dist-info/WHEEL,sha256=5zIMuEcQBP51tQKLCGHvUIyt6WyXoCqqHwQqkDIOwdw,110
|
|
408
|
+
warp_lang-1.0.0b6.dist-info/top_level.txt,sha256=pGqQ4zb6p2PGWj4LRHmD9iUQdiR7ZJCpvHTYlsIp7BI,44
|
|
409
|
+
warp_lang-1.0.0b6.dist-info/RECORD,,
|
|
@@ -1,40 +0,0 @@
|
|
|
1
|
-
# Copyright (c) 2022 NVIDIA CORPORATION. All rights reserved.
|
|
2
|
-
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
3
|
-
# and proprietary rights in and to this software, related documentation
|
|
4
|
-
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
5
|
-
# distribution of this software and related documentation without an express
|
|
6
|
-
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
7
|
-
|
|
8
|
-
import warp as wp
|
|
9
|
-
import os
|
|
10
|
-
|
|
11
|
-
example_dir = os.path.dirname(os.path.realpath(__file__))
|
|
12
|
-
|
|
13
|
-
# set default cache directory before wp.init()
|
|
14
|
-
wp.config.kernel_cache_dir = os.path.join(example_dir, "tmp", "warpcache1")
|
|
15
|
-
|
|
16
|
-
wp.init()
|
|
17
|
-
|
|
18
|
-
print("+++ Current cache directory: ", wp.config.kernel_cache_dir)
|
|
19
|
-
|
|
20
|
-
# change cache directory after wp.init()
|
|
21
|
-
wp.build.init_kernel_cache(os.path.join(example_dir, "tmp", "warpcache2"))
|
|
22
|
-
|
|
23
|
-
print("+++ Current cache directory: ", wp.config.kernel_cache_dir)
|
|
24
|
-
|
|
25
|
-
# clear kernel cache (forces fresh kernel builds every time)
|
|
26
|
-
wp.build.clear_kernel_cache()
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
@wp.kernel
|
|
30
|
-
def basic(x: wp.array(dtype=float)):
|
|
31
|
-
tid = wp.tid()
|
|
32
|
-
x[tid] = float(tid)
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
device = "cpu"
|
|
36
|
-
n = 10
|
|
37
|
-
x = wp.zeros(n, dtype=float, device=device)
|
|
38
|
-
|
|
39
|
-
wp.launch(kernel=basic, dim=n, inputs=[x], device=device)
|
|
40
|
-
print(x.numpy())
|
examples/example_multigpu.py
DELETED
|
@@ -1,54 +0,0 @@
|
|
|
1
|
-
# Copyright (c) 2022 NVIDIA CORPORATION. All rights reserved.
|
|
2
|
-
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
3
|
-
# and proprietary rights in and to this software, related documentation
|
|
4
|
-
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
5
|
-
# distribution of this software and related documentation without an express
|
|
6
|
-
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
7
|
-
|
|
8
|
-
###########################################################################
|
|
9
|
-
# Example Multi-GPU
|
|
10
|
-
#
|
|
11
|
-
# A basic example that shows how to allocate arrays and launch kernels
|
|
12
|
-
# on all available CUDA devices.
|
|
13
|
-
#
|
|
14
|
-
###########################################################################
|
|
15
|
-
|
|
16
|
-
import warp as wp
|
|
17
|
-
|
|
18
|
-
wp.init()
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@wp.kernel
|
|
22
|
-
def inc(a: wp.array(dtype=float)):
|
|
23
|
-
tid = wp.tid()
|
|
24
|
-
a[tid] = a[tid] + 1.0
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
# get all CUDA devices
|
|
28
|
-
devices = wp.get_cuda_devices()
|
|
29
|
-
device_count = len(devices)
|
|
30
|
-
|
|
31
|
-
# number of launches
|
|
32
|
-
iters = 1000
|
|
33
|
-
|
|
34
|
-
# list of arrays, one per device
|
|
35
|
-
arrs = []
|
|
36
|
-
|
|
37
|
-
# loop over all devices
|
|
38
|
-
for device in devices:
|
|
39
|
-
# use a ScopedDevice to set the target device
|
|
40
|
-
with wp.ScopedDevice(device):
|
|
41
|
-
# allocate array
|
|
42
|
-
a = wp.zeros(250 * 1024 * 1024, dtype=float)
|
|
43
|
-
arrs.append(a)
|
|
44
|
-
|
|
45
|
-
# launch kernels
|
|
46
|
-
for _ in range(iters):
|
|
47
|
-
wp.launch(inc, dim=a.size, inputs=[a])
|
|
48
|
-
|
|
49
|
-
# synchronize all devices
|
|
50
|
-
wp.synchronize()
|
|
51
|
-
|
|
52
|
-
# print results
|
|
53
|
-
for i in range(device_count):
|
|
54
|
-
print(f"{arrs[i].device} -> {arrs[i].numpy()}")
|
examples/example_struct.py
DELETED
|
@@ -1,65 +0,0 @@
|
|
|
1
|
-
# Copyright (c) 2022 NVIDIA CORPORATION. All rights reserved.
|
|
2
|
-
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
3
|
-
# and proprietary rights in and to this software, related documentation
|
|
4
|
-
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
5
|
-
# distribution of this software and related documentation without an express
|
|
6
|
-
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
7
|
-
|
|
8
|
-
###########################################################################
|
|
9
|
-
# Example Struct
|
|
10
|
-
#
|
|
11
|
-
# Shows how to define a custom struct and use it during gradient computation
|
|
12
|
-
#
|
|
13
|
-
###########################################################################
|
|
14
|
-
|
|
15
|
-
import numpy as np
|
|
16
|
-
import warp as wp
|
|
17
|
-
|
|
18
|
-
wp.init()
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@wp.struct
|
|
22
|
-
class TestStruct:
|
|
23
|
-
x: wp.vec3
|
|
24
|
-
a: wp.array(dtype=wp.vec3)
|
|
25
|
-
b: wp.array(dtype=wp.vec3)
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
@wp.kernel
|
|
29
|
-
def test_kernel(s: TestStruct):
|
|
30
|
-
tid = wp.tid()
|
|
31
|
-
|
|
32
|
-
s.b[tid] = s.a[tid] + s.x
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
@wp.kernel
|
|
36
|
-
def loss_kernel(s: TestStruct, loss: wp.array(dtype=float)):
|
|
37
|
-
tid = wp.tid()
|
|
38
|
-
|
|
39
|
-
v = s.b[tid]
|
|
40
|
-
wp.atomic_add(loss, 0, float(tid + 1) * (v[0] + 2.0 * v[1] + 3.0 * v[2]))
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
# create struct
|
|
44
|
-
ts = TestStruct()
|
|
45
|
-
|
|
46
|
-
# set members
|
|
47
|
-
ts.x = wp.vec3(1.0, 2.0, 3.0)
|
|
48
|
-
ts.a = wp.array(
|
|
49
|
-
np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]),
|
|
50
|
-
dtype=wp.vec3,
|
|
51
|
-
requires_grad=True,
|
|
52
|
-
)
|
|
53
|
-
ts.b = wp.zeros(2, dtype=wp.vec3, requires_grad=True)
|
|
54
|
-
|
|
55
|
-
loss = wp.zeros(1, dtype=float, requires_grad=True)
|
|
56
|
-
|
|
57
|
-
tape = wp.Tape()
|
|
58
|
-
with tape:
|
|
59
|
-
wp.launch(test_kernel, dim=2, inputs=[ts])
|
|
60
|
-
wp.launch(loss_kernel, dim=2, inputs=[ts, loss])
|
|
61
|
-
|
|
62
|
-
tape.backward(loss)
|
|
63
|
-
|
|
64
|
-
print(loss)
|
|
65
|
-
print(tape.gradients[ts].a)
|