warp-lang 1.7.0__py3-none-manylinux_2_28_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.
Potentially problematic release.
This version of warp-lang might be problematic. Click here for more details.
- warp/__init__.py +139 -0
- warp/__init__.pyi +1 -0
- warp/autograd.py +1142 -0
- warp/bin/warp-clang.so +0 -0
- warp/bin/warp.so +0 -0
- warp/build.py +557 -0
- warp/build_dll.py +405 -0
- warp/builtins.py +6855 -0
- warp/codegen.py +3969 -0
- warp/config.py +158 -0
- warp/constants.py +57 -0
- warp/context.py +6812 -0
- warp/dlpack.py +462 -0
- warp/examples/__init__.py +24 -0
- warp/examples/assets/bear.usd +0 -0
- warp/examples/assets/bunny.usd +0 -0
- warp/examples/assets/cartpole.urdf +110 -0
- warp/examples/assets/crazyflie.usd +0 -0
- warp/examples/assets/cube.usd +0 -0
- warp/examples/assets/nonuniform.usd +0 -0
- warp/examples/assets/nv_ant.xml +92 -0
- warp/examples/assets/nv_humanoid.xml +183 -0
- warp/examples/assets/nvidia_logo.png +0 -0
- warp/examples/assets/pixel.jpg +0 -0
- warp/examples/assets/quadruped.urdf +268 -0
- warp/examples/assets/rocks.nvdb +0 -0
- warp/examples/assets/rocks.usd +0 -0
- warp/examples/assets/sphere.usd +0 -0
- warp/examples/assets/square_cloth.usd +0 -0
- warp/examples/benchmarks/benchmark_api.py +389 -0
- warp/examples/benchmarks/benchmark_cloth.py +296 -0
- warp/examples/benchmarks/benchmark_cloth_cupy.py +96 -0
- warp/examples/benchmarks/benchmark_cloth_jax.py +105 -0
- warp/examples/benchmarks/benchmark_cloth_numba.py +161 -0
- warp/examples/benchmarks/benchmark_cloth_numpy.py +85 -0
- warp/examples/benchmarks/benchmark_cloth_paddle.py +94 -0
- warp/examples/benchmarks/benchmark_cloth_pytorch.py +94 -0
- warp/examples/benchmarks/benchmark_cloth_taichi.py +120 -0
- warp/examples/benchmarks/benchmark_cloth_warp.py +153 -0
- warp/examples/benchmarks/benchmark_gemm.py +164 -0
- warp/examples/benchmarks/benchmark_interop_paddle.py +166 -0
- warp/examples/benchmarks/benchmark_interop_torch.py +166 -0
- warp/examples/benchmarks/benchmark_launches.py +301 -0
- warp/examples/benchmarks/benchmark_tile_load_store.py +103 -0
- warp/examples/browse.py +37 -0
- warp/examples/core/example_cupy.py +86 -0
- warp/examples/core/example_dem.py +241 -0
- warp/examples/core/example_fluid.py +299 -0
- warp/examples/core/example_graph_capture.py +150 -0
- warp/examples/core/example_marching_cubes.py +194 -0
- warp/examples/core/example_mesh.py +180 -0
- warp/examples/core/example_mesh_intersect.py +211 -0
- warp/examples/core/example_nvdb.py +182 -0
- warp/examples/core/example_raycast.py +111 -0
- warp/examples/core/example_raymarch.py +205 -0
- warp/examples/core/example_render_opengl.py +193 -0
- warp/examples/core/example_sample_mesh.py +300 -0
- warp/examples/core/example_sph.py +411 -0
- warp/examples/core/example_torch.py +211 -0
- warp/examples/core/example_wave.py +269 -0
- warp/examples/fem/example_adaptive_grid.py +286 -0
- warp/examples/fem/example_apic_fluid.py +423 -0
- warp/examples/fem/example_burgers.py +261 -0
- warp/examples/fem/example_convection_diffusion.py +178 -0
- warp/examples/fem/example_convection_diffusion_dg.py +204 -0
- warp/examples/fem/example_deformed_geometry.py +172 -0
- warp/examples/fem/example_diffusion.py +196 -0
- warp/examples/fem/example_diffusion_3d.py +225 -0
- warp/examples/fem/example_diffusion_mgpu.py +220 -0
- warp/examples/fem/example_distortion_energy.py +228 -0
- warp/examples/fem/example_magnetostatics.py +240 -0
- warp/examples/fem/example_mixed_elasticity.py +291 -0
- warp/examples/fem/example_navier_stokes.py +261 -0
- warp/examples/fem/example_nonconforming_contact.py +298 -0
- warp/examples/fem/example_stokes.py +213 -0
- warp/examples/fem/example_stokes_transfer.py +262 -0
- warp/examples/fem/example_streamlines.py +352 -0
- warp/examples/fem/utils.py +1000 -0
- warp/examples/interop/example_jax_callable.py +116 -0
- warp/examples/interop/example_jax_ffi_callback.py +132 -0
- warp/examples/interop/example_jax_kernel.py +205 -0
- warp/examples/optim/example_bounce.py +266 -0
- warp/examples/optim/example_cloth_throw.py +228 -0
- warp/examples/optim/example_diffray.py +561 -0
- warp/examples/optim/example_drone.py +870 -0
- warp/examples/optim/example_fluid_checkpoint.py +497 -0
- warp/examples/optim/example_inverse_kinematics.py +182 -0
- warp/examples/optim/example_inverse_kinematics_torch.py +191 -0
- warp/examples/optim/example_softbody_properties.py +400 -0
- warp/examples/optim/example_spring_cage.py +245 -0
- warp/examples/optim/example_trajectory.py +227 -0
- warp/examples/sim/example_cartpole.py +143 -0
- warp/examples/sim/example_cloth.py +225 -0
- warp/examples/sim/example_cloth_self_contact.py +322 -0
- warp/examples/sim/example_granular.py +130 -0
- warp/examples/sim/example_granular_collision_sdf.py +202 -0
- warp/examples/sim/example_jacobian_ik.py +244 -0
- warp/examples/sim/example_particle_chain.py +124 -0
- warp/examples/sim/example_quadruped.py +203 -0
- warp/examples/sim/example_rigid_chain.py +203 -0
- warp/examples/sim/example_rigid_contact.py +195 -0
- warp/examples/sim/example_rigid_force.py +133 -0
- warp/examples/sim/example_rigid_gyroscopic.py +115 -0
- warp/examples/sim/example_rigid_soft_contact.py +140 -0
- warp/examples/sim/example_soft_body.py +196 -0
- warp/examples/tile/example_tile_cholesky.py +87 -0
- warp/examples/tile/example_tile_convolution.py +66 -0
- warp/examples/tile/example_tile_fft.py +55 -0
- warp/examples/tile/example_tile_filtering.py +113 -0
- warp/examples/tile/example_tile_matmul.py +85 -0
- warp/examples/tile/example_tile_mlp.py +383 -0
- warp/examples/tile/example_tile_nbody.py +199 -0
- warp/examples/tile/example_tile_walker.py +327 -0
- warp/fabric.py +355 -0
- warp/fem/__init__.py +106 -0
- warp/fem/adaptivity.py +508 -0
- warp/fem/cache.py +572 -0
- warp/fem/dirichlet.py +202 -0
- warp/fem/domain.py +411 -0
- warp/fem/field/__init__.py +125 -0
- warp/fem/field/field.py +619 -0
- warp/fem/field/nodal_field.py +326 -0
- warp/fem/field/restriction.py +37 -0
- warp/fem/field/virtual.py +848 -0
- warp/fem/geometry/__init__.py +32 -0
- warp/fem/geometry/adaptive_nanogrid.py +857 -0
- warp/fem/geometry/closest_point.py +84 -0
- warp/fem/geometry/deformed_geometry.py +221 -0
- warp/fem/geometry/element.py +776 -0
- warp/fem/geometry/geometry.py +362 -0
- warp/fem/geometry/grid_2d.py +392 -0
- warp/fem/geometry/grid_3d.py +452 -0
- warp/fem/geometry/hexmesh.py +911 -0
- warp/fem/geometry/nanogrid.py +571 -0
- warp/fem/geometry/partition.py +389 -0
- warp/fem/geometry/quadmesh.py +663 -0
- warp/fem/geometry/tetmesh.py +855 -0
- warp/fem/geometry/trimesh.py +806 -0
- warp/fem/integrate.py +2335 -0
- warp/fem/linalg.py +419 -0
- warp/fem/operator.py +293 -0
- warp/fem/polynomial.py +229 -0
- warp/fem/quadrature/__init__.py +17 -0
- warp/fem/quadrature/pic_quadrature.py +299 -0
- warp/fem/quadrature/quadrature.py +591 -0
- warp/fem/space/__init__.py +228 -0
- warp/fem/space/basis_function_space.py +468 -0
- warp/fem/space/basis_space.py +667 -0
- warp/fem/space/dof_mapper.py +251 -0
- warp/fem/space/function_space.py +309 -0
- warp/fem/space/grid_2d_function_space.py +177 -0
- warp/fem/space/grid_3d_function_space.py +227 -0
- warp/fem/space/hexmesh_function_space.py +257 -0
- warp/fem/space/nanogrid_function_space.py +201 -0
- warp/fem/space/partition.py +367 -0
- warp/fem/space/quadmesh_function_space.py +223 -0
- warp/fem/space/restriction.py +179 -0
- warp/fem/space/shape/__init__.py +143 -0
- warp/fem/space/shape/cube_shape_function.py +1105 -0
- warp/fem/space/shape/shape_function.py +133 -0
- warp/fem/space/shape/square_shape_function.py +926 -0
- warp/fem/space/shape/tet_shape_function.py +834 -0
- warp/fem/space/shape/triangle_shape_function.py +672 -0
- warp/fem/space/tetmesh_function_space.py +271 -0
- warp/fem/space/topology.py +424 -0
- warp/fem/space/trimesh_function_space.py +194 -0
- warp/fem/types.py +99 -0
- warp/fem/utils.py +420 -0
- warp/jax.py +187 -0
- warp/jax_experimental/__init__.py +16 -0
- warp/jax_experimental/custom_call.py +351 -0
- warp/jax_experimental/ffi.py +698 -0
- warp/jax_experimental/xla_ffi.py +602 -0
- warp/math.py +244 -0
- warp/native/array.h +1145 -0
- warp/native/builtin.h +1800 -0
- warp/native/bvh.cpp +492 -0
- warp/native/bvh.cu +791 -0
- warp/native/bvh.h +554 -0
- warp/native/clang/clang.cpp +536 -0
- warp/native/coloring.cpp +613 -0
- warp/native/crt.cpp +51 -0
- warp/native/crt.h +362 -0
- warp/native/cuda_crt.h +1058 -0
- warp/native/cuda_util.cpp +646 -0
- warp/native/cuda_util.h +307 -0
- warp/native/error.cpp +77 -0
- warp/native/error.h +36 -0
- warp/native/exports.h +1878 -0
- warp/native/fabric.h +245 -0
- warp/native/hashgrid.cpp +311 -0
- warp/native/hashgrid.cu +87 -0
- warp/native/hashgrid.h +240 -0
- warp/native/initializer_array.h +41 -0
- warp/native/intersect.h +1230 -0
- warp/native/intersect_adj.h +375 -0
- warp/native/intersect_tri.h +339 -0
- warp/native/marching.cpp +19 -0
- warp/native/marching.cu +514 -0
- warp/native/marching.h +19 -0
- warp/native/mat.h +2220 -0
- warp/native/mathdx.cpp +87 -0
- warp/native/matnn.h +343 -0
- warp/native/mesh.cpp +266 -0
- warp/native/mesh.cu +404 -0
- warp/native/mesh.h +1980 -0
- warp/native/nanovdb/GridHandle.h +366 -0
- warp/native/nanovdb/HostBuffer.h +590 -0
- warp/native/nanovdb/NanoVDB.h +6624 -0
- warp/native/nanovdb/PNanoVDB.h +3390 -0
- warp/native/noise.h +859 -0
- warp/native/quat.h +1371 -0
- warp/native/rand.h +342 -0
- warp/native/range.h +139 -0
- warp/native/reduce.cpp +174 -0
- warp/native/reduce.cu +364 -0
- warp/native/runlength_encode.cpp +79 -0
- warp/native/runlength_encode.cu +61 -0
- warp/native/scan.cpp +47 -0
- warp/native/scan.cu +53 -0
- warp/native/scan.h +23 -0
- warp/native/solid_angle.h +466 -0
- warp/native/sort.cpp +251 -0
- warp/native/sort.cu +277 -0
- warp/native/sort.h +33 -0
- warp/native/sparse.cpp +378 -0
- warp/native/sparse.cu +524 -0
- warp/native/spatial.h +657 -0
- warp/native/svd.h +702 -0
- warp/native/temp_buffer.h +46 -0
- warp/native/tile.h +2584 -0
- warp/native/tile_reduce.h +264 -0
- warp/native/vec.h +1426 -0
- warp/native/volume.cpp +501 -0
- warp/native/volume.cu +67 -0
- warp/native/volume.h +969 -0
- warp/native/volume_builder.cu +477 -0
- warp/native/volume_builder.h +52 -0
- warp/native/volume_impl.h +70 -0
- warp/native/warp.cpp +1082 -0
- warp/native/warp.cu +3636 -0
- warp/native/warp.h +381 -0
- warp/optim/__init__.py +17 -0
- warp/optim/adam.py +163 -0
- warp/optim/linear.py +1137 -0
- warp/optim/sgd.py +112 -0
- warp/paddle.py +407 -0
- warp/render/__init__.py +18 -0
- warp/render/render_opengl.py +3518 -0
- warp/render/render_usd.py +784 -0
- warp/render/utils.py +160 -0
- warp/sim/__init__.py +65 -0
- warp/sim/articulation.py +793 -0
- warp/sim/collide.py +2395 -0
- warp/sim/graph_coloring.py +300 -0
- warp/sim/import_mjcf.py +790 -0
- warp/sim/import_snu.py +227 -0
- warp/sim/import_urdf.py +579 -0
- warp/sim/import_usd.py +894 -0
- warp/sim/inertia.py +324 -0
- warp/sim/integrator.py +242 -0
- warp/sim/integrator_euler.py +1997 -0
- warp/sim/integrator_featherstone.py +2101 -0
- warp/sim/integrator_vbd.py +2048 -0
- warp/sim/integrator_xpbd.py +3292 -0
- warp/sim/model.py +4791 -0
- warp/sim/particles.py +121 -0
- warp/sim/render.py +427 -0
- warp/sim/utils.py +428 -0
- warp/sparse.py +2057 -0
- warp/stubs.py +3333 -0
- warp/tape.py +1203 -0
- warp/tests/__init__.py +1 -0
- warp/tests/__main__.py +4 -0
- warp/tests/assets/curlnoise_golden.npy +0 -0
- warp/tests/assets/mlp_golden.npy +0 -0
- warp/tests/assets/pixel.npy +0 -0
- warp/tests/assets/pnoise_golden.npy +0 -0
- warp/tests/assets/spiky.usd +0 -0
- warp/tests/assets/test_grid.nvdb +0 -0
- warp/tests/assets/test_index_grid.nvdb +0 -0
- warp/tests/assets/test_int32_grid.nvdb +0 -0
- warp/tests/assets/test_vec_grid.nvdb +0 -0
- warp/tests/assets/torus.nvdb +0 -0
- warp/tests/assets/torus.usda +105 -0
- warp/tests/aux_test_class_kernel.py +34 -0
- warp/tests/aux_test_compile_consts_dummy.py +18 -0
- warp/tests/aux_test_conditional_unequal_types_kernels.py +29 -0
- warp/tests/aux_test_dependent.py +29 -0
- warp/tests/aux_test_grad_customs.py +29 -0
- warp/tests/aux_test_instancing_gc.py +26 -0
- warp/tests/aux_test_module_unload.py +23 -0
- warp/tests/aux_test_name_clash1.py +40 -0
- warp/tests/aux_test_name_clash2.py +40 -0
- warp/tests/aux_test_reference.py +9 -0
- warp/tests/aux_test_reference_reference.py +8 -0
- warp/tests/aux_test_square.py +16 -0
- warp/tests/aux_test_unresolved_func.py +22 -0
- warp/tests/aux_test_unresolved_symbol.py +22 -0
- warp/tests/cuda/__init__.py +0 -0
- warp/tests/cuda/test_async.py +676 -0
- warp/tests/cuda/test_ipc.py +124 -0
- warp/tests/cuda/test_mempool.py +233 -0
- warp/tests/cuda/test_multigpu.py +169 -0
- warp/tests/cuda/test_peer.py +139 -0
- warp/tests/cuda/test_pinned.py +84 -0
- warp/tests/cuda/test_streams.py +634 -0
- warp/tests/geometry/__init__.py +0 -0
- warp/tests/geometry/test_bvh.py +200 -0
- warp/tests/geometry/test_hash_grid.py +221 -0
- warp/tests/geometry/test_marching_cubes.py +74 -0
- warp/tests/geometry/test_mesh.py +316 -0
- warp/tests/geometry/test_mesh_query_aabb.py +399 -0
- warp/tests/geometry/test_mesh_query_point.py +932 -0
- warp/tests/geometry/test_mesh_query_ray.py +311 -0
- warp/tests/geometry/test_volume.py +1103 -0
- warp/tests/geometry/test_volume_write.py +346 -0
- warp/tests/interop/__init__.py +0 -0
- warp/tests/interop/test_dlpack.py +729 -0
- warp/tests/interop/test_jax.py +371 -0
- warp/tests/interop/test_paddle.py +800 -0
- warp/tests/interop/test_torch.py +1001 -0
- warp/tests/run_coverage_serial.py +39 -0
- warp/tests/sim/__init__.py +0 -0
- warp/tests/sim/disabled_kinematics.py +244 -0
- warp/tests/sim/flaky_test_sim_grad.py +290 -0
- warp/tests/sim/test_collision.py +604 -0
- warp/tests/sim/test_coloring.py +258 -0
- warp/tests/sim/test_model.py +224 -0
- warp/tests/sim/test_sim_grad_bounce_linear.py +212 -0
- warp/tests/sim/test_sim_kinematics.py +98 -0
- warp/tests/sim/test_vbd.py +597 -0
- warp/tests/test_adam.py +163 -0
- warp/tests/test_arithmetic.py +1096 -0
- warp/tests/test_array.py +2972 -0
- warp/tests/test_array_reduce.py +156 -0
- warp/tests/test_assert.py +250 -0
- warp/tests/test_atomic.py +153 -0
- warp/tests/test_bool.py +220 -0
- warp/tests/test_builtins_resolution.py +1298 -0
- warp/tests/test_closest_point_edge_edge.py +327 -0
- warp/tests/test_codegen.py +810 -0
- warp/tests/test_codegen_instancing.py +1495 -0
- warp/tests/test_compile_consts.py +215 -0
- warp/tests/test_conditional.py +252 -0
- warp/tests/test_context.py +42 -0
- warp/tests/test_copy.py +238 -0
- warp/tests/test_ctypes.py +638 -0
- warp/tests/test_dense.py +73 -0
- warp/tests/test_devices.py +97 -0
- warp/tests/test_examples.py +482 -0
- warp/tests/test_fabricarray.py +996 -0
- warp/tests/test_fast_math.py +74 -0
- warp/tests/test_fem.py +2003 -0
- warp/tests/test_fp16.py +136 -0
- warp/tests/test_func.py +454 -0
- warp/tests/test_future_annotations.py +98 -0
- warp/tests/test_generics.py +656 -0
- warp/tests/test_grad.py +893 -0
- warp/tests/test_grad_customs.py +339 -0
- warp/tests/test_grad_debug.py +341 -0
- warp/tests/test_implicit_init.py +411 -0
- warp/tests/test_import.py +45 -0
- warp/tests/test_indexedarray.py +1140 -0
- warp/tests/test_intersect.py +73 -0
- warp/tests/test_iter.py +76 -0
- warp/tests/test_large.py +177 -0
- warp/tests/test_launch.py +411 -0
- warp/tests/test_lerp.py +151 -0
- warp/tests/test_linear_solvers.py +193 -0
- warp/tests/test_lvalue.py +427 -0
- warp/tests/test_mat.py +2089 -0
- warp/tests/test_mat_lite.py +122 -0
- warp/tests/test_mat_scalar_ops.py +2913 -0
- warp/tests/test_math.py +178 -0
- warp/tests/test_mlp.py +282 -0
- warp/tests/test_module_hashing.py +258 -0
- warp/tests/test_modules_lite.py +44 -0
- warp/tests/test_noise.py +252 -0
- warp/tests/test_operators.py +299 -0
- warp/tests/test_options.py +129 -0
- warp/tests/test_overwrite.py +551 -0
- warp/tests/test_print.py +339 -0
- warp/tests/test_quat.py +2315 -0
- warp/tests/test_rand.py +339 -0
- warp/tests/test_reload.py +302 -0
- warp/tests/test_rounding.py +185 -0
- warp/tests/test_runlength_encode.py +196 -0
- warp/tests/test_scalar_ops.py +105 -0
- warp/tests/test_smoothstep.py +108 -0
- warp/tests/test_snippet.py +318 -0
- warp/tests/test_sparse.py +582 -0
- warp/tests/test_spatial.py +2229 -0
- warp/tests/test_special_values.py +361 -0
- warp/tests/test_static.py +592 -0
- warp/tests/test_struct.py +734 -0
- warp/tests/test_tape.py +204 -0
- warp/tests/test_transient_module.py +93 -0
- warp/tests/test_triangle_closest_point.py +145 -0
- warp/tests/test_types.py +562 -0
- warp/tests/test_utils.py +588 -0
- warp/tests/test_vec.py +1487 -0
- warp/tests/test_vec_lite.py +80 -0
- warp/tests/test_vec_scalar_ops.py +2327 -0
- warp/tests/test_verify_fp.py +100 -0
- warp/tests/tile/__init__.py +0 -0
- warp/tests/tile/test_tile.py +780 -0
- warp/tests/tile/test_tile_load.py +407 -0
- warp/tests/tile/test_tile_mathdx.py +208 -0
- warp/tests/tile/test_tile_mlp.py +402 -0
- warp/tests/tile/test_tile_reduce.py +447 -0
- warp/tests/tile/test_tile_shared_memory.py +247 -0
- warp/tests/tile/test_tile_view.py +173 -0
- warp/tests/unittest_serial.py +47 -0
- warp/tests/unittest_suites.py +427 -0
- warp/tests/unittest_utils.py +468 -0
- warp/tests/walkthrough_debug.py +93 -0
- warp/thirdparty/__init__.py +0 -0
- warp/thirdparty/appdirs.py +598 -0
- warp/thirdparty/dlpack.py +145 -0
- warp/thirdparty/unittest_parallel.py +570 -0
- warp/torch.py +391 -0
- warp/types.py +5230 -0
- warp/utils.py +1137 -0
- warp_lang-1.7.0.dist-info/METADATA +516 -0
- warp_lang-1.7.0.dist-info/RECORD +429 -0
- warp_lang-1.7.0.dist-info/WHEEL +5 -0
- warp_lang-1.7.0.dist-info/licenses/LICENSE.md +202 -0
- warp_lang-1.7.0.dist-info/top_level.txt +1 -0
warp/native/cuda_crt.h
ADDED
|
@@ -0,0 +1,1058 @@
|
|
|
1
|
+
/*
|
|
2
|
+
* SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
3
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
4
|
+
*
|
|
5
|
+
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
+
* you may not use this file except in compliance with the License.
|
|
7
|
+
* You may obtain a copy of the License at
|
|
8
|
+
*
|
|
9
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
+
*
|
|
11
|
+
* Unless required by applicable law or agreed to in writing, software
|
|
12
|
+
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
+
* See the License for the specific language governing permissions and
|
|
15
|
+
* limitations under the License.
|
|
16
|
+
*/
|
|
17
|
+
|
|
18
|
+
#pragma once
|
|
19
|
+
|
|
20
|
+
// Attributes otherwise defined by CUDA's crt/host_defines.h
|
|
21
|
+
#define __constant__ __attribute__((constant))
|
|
22
|
+
#define __device__ __attribute__((device))
|
|
23
|
+
#define __global__ __attribute__((global))
|
|
24
|
+
#define __host__ __attribute__((host))
|
|
25
|
+
#define __shared__ __attribute__((shared))
|
|
26
|
+
#define __forceinline__ __attribute__((always_inline))
|
|
27
|
+
|
|
28
|
+
using size_t = unsigned long long;
|
|
29
|
+
|
|
30
|
+
// stdio.h
|
|
31
|
+
extern "C" __device__ int printf(const char* format, ... );
|
|
32
|
+
|
|
33
|
+
// assert.h
|
|
34
|
+
extern __host__ __device__ void __assertfail(const char * __assertion,
|
|
35
|
+
const char *__file,
|
|
36
|
+
unsigned int __line,
|
|
37
|
+
const char *__function,
|
|
38
|
+
size_t charsize);
|
|
39
|
+
|
|
40
|
+
#if defined(NDEBUG)
|
|
41
|
+
#define assert(e) (static_cast<void>(0))
|
|
42
|
+
#else /* !NDEBUG */
|
|
43
|
+
#define __ASSERT_STR_HELPER(x) #x
|
|
44
|
+
#define assert(e) ((e) ? static_cast<void>(0)\
|
|
45
|
+
: __assertfail(__ASSERT_STR_HELPER(e), __FILE__,\
|
|
46
|
+
__LINE__, __PRETTY_FUNCTION__,\
|
|
47
|
+
sizeof(char)))
|
|
48
|
+
#endif
|
|
49
|
+
|
|
50
|
+
#define __device_forceinline__ static __device__ inline __forceinline__
|
|
51
|
+
|
|
52
|
+
// Implementations of CUDA builtin variables
|
|
53
|
+
struct __threadIdx_t
|
|
54
|
+
{
|
|
55
|
+
__declspec(property(get = get_x)) unsigned int x;
|
|
56
|
+
__declspec(property(get = get_y)) unsigned int y;
|
|
57
|
+
__declspec(property(get = get_z)) unsigned int z;
|
|
58
|
+
|
|
59
|
+
__device_forceinline__ unsigned int get_x() { return __nvvm_read_ptx_sreg_tid_x(); }
|
|
60
|
+
__device_forceinline__ unsigned int get_y() { return __nvvm_read_ptx_sreg_tid_y(); }
|
|
61
|
+
__device_forceinline__ unsigned int get_z() { return __nvvm_read_ptx_sreg_tid_z(); }
|
|
62
|
+
};
|
|
63
|
+
|
|
64
|
+
struct __blockIdx_t
|
|
65
|
+
{
|
|
66
|
+
__declspec(property(get = get_x)) unsigned int x;
|
|
67
|
+
__declspec(property(get = get_y)) unsigned int y;
|
|
68
|
+
__declspec(property(get = get_z)) unsigned int z;
|
|
69
|
+
|
|
70
|
+
__device_forceinline__ unsigned int get_x() { return __nvvm_read_ptx_sreg_ctaid_x(); }
|
|
71
|
+
__device_forceinline__ unsigned int get_y() { return __nvvm_read_ptx_sreg_ctaid_y(); }
|
|
72
|
+
__device_forceinline__ unsigned int get_z() { return __nvvm_read_ptx_sreg_ctaid_z(); }
|
|
73
|
+
};
|
|
74
|
+
|
|
75
|
+
struct __blockDim_t
|
|
76
|
+
{
|
|
77
|
+
__declspec(property(get = get_x)) unsigned int x;
|
|
78
|
+
__declspec(property(get = get_y)) unsigned int y;
|
|
79
|
+
__declspec(property(get = get_z)) unsigned int z;
|
|
80
|
+
|
|
81
|
+
__device_forceinline__ unsigned int get_x() { return __nvvm_read_ptx_sreg_ntid_x(); }
|
|
82
|
+
__device_forceinline__ unsigned int get_y() { return __nvvm_read_ptx_sreg_ntid_y(); }
|
|
83
|
+
__device_forceinline__ unsigned int get_z() { return __nvvm_read_ptx_sreg_ntid_z(); }
|
|
84
|
+
};
|
|
85
|
+
|
|
86
|
+
struct __gridDim_t
|
|
87
|
+
{
|
|
88
|
+
__declspec(property(get = get_x)) unsigned int x;
|
|
89
|
+
__declspec(property(get = get_y)) unsigned int y;
|
|
90
|
+
__declspec(property(get = get_z)) unsigned int z;
|
|
91
|
+
|
|
92
|
+
__device_forceinline__ unsigned int get_x() { return __nvvm_read_ptx_sreg_nctaid_x(); }
|
|
93
|
+
__device_forceinline__ unsigned int get_y() { return __nvvm_read_ptx_sreg_nctaid_y(); }
|
|
94
|
+
__device_forceinline__ unsigned int get_z() { return __nvvm_read_ptx_sreg_nctaid_z(); }
|
|
95
|
+
};
|
|
96
|
+
|
|
97
|
+
extern const __device__ __threadIdx_t threadIdx;
|
|
98
|
+
extern const __device__ __blockIdx_t blockIdx;
|
|
99
|
+
extern const __device__ __blockDim_t blockDim;
|
|
100
|
+
extern const __device__ __gridDim_t gridDim;
|
|
101
|
+
|
|
102
|
+
// Forward declarations of libdevice functions
|
|
103
|
+
extern "C" {
|
|
104
|
+
|
|
105
|
+
__device__ int __nv_abs(int a);
|
|
106
|
+
__device__ double __nv_acos(double a);
|
|
107
|
+
__device__ float __nv_acosf(float a);
|
|
108
|
+
__device__ double __nv_acosh(double a);
|
|
109
|
+
__device__ float __nv_acoshf(float a);
|
|
110
|
+
__device__ double __nv_asin(double a);
|
|
111
|
+
__device__ float __nv_asinf(float a);
|
|
112
|
+
__device__ double __nv_asinh(double a);
|
|
113
|
+
__device__ float __nv_asinhf(float a);
|
|
114
|
+
__device__ double __nv_atan2(double a, double b);
|
|
115
|
+
__device__ float __nv_atan2f(float a, float b);
|
|
116
|
+
__device__ double __nv_atan(double a);
|
|
117
|
+
__device__ float __nv_atanf(float a);
|
|
118
|
+
__device__ double __nv_atanh(double a);
|
|
119
|
+
__device__ float __nv_atanhf(float a);
|
|
120
|
+
__device__ int __nv_brev(int a);
|
|
121
|
+
__device__ long long __nv_brevll(long long a);
|
|
122
|
+
__device__ int __nv_byte_perm(int a, int b, int c);
|
|
123
|
+
__device__ double __nv_cbrt(double a);
|
|
124
|
+
__device__ float __nv_cbrtf(float a);
|
|
125
|
+
__device__ double __nv_ceil(double a);
|
|
126
|
+
__device__ float __nv_ceilf(float a);
|
|
127
|
+
__device__ int __nv_clz(int a);
|
|
128
|
+
__device__ int __nv_clzll(long long a);
|
|
129
|
+
__device__ double __nv_copysign(double a, double b);
|
|
130
|
+
__device__ float __nv_copysignf(float a, float b);
|
|
131
|
+
__device__ double __nv_cos(double a);
|
|
132
|
+
__device__ float __nv_cosf(float a);
|
|
133
|
+
__device__ double __nv_cosh(double a);
|
|
134
|
+
__device__ float __nv_coshf(float a);
|
|
135
|
+
__device__ double __nv_cospi(double a);
|
|
136
|
+
__device__ float __nv_cospif(float a);
|
|
137
|
+
__device__ double __nv_cyl_bessel_i0(double a);
|
|
138
|
+
__device__ float __nv_cyl_bessel_i0f(float a);
|
|
139
|
+
__device__ double __nv_cyl_bessel_i1(double a);
|
|
140
|
+
__device__ float __nv_cyl_bessel_i1f(float a);
|
|
141
|
+
__device__ double __nv_dadd_rd(double a, double b);
|
|
142
|
+
__device__ double __nv_dadd_rn(double a, double b);
|
|
143
|
+
__device__ double __nv_dadd_ru(double a, double b);
|
|
144
|
+
__device__ double __nv_dadd_rz(double a, double b);
|
|
145
|
+
__device__ double __nv_ddiv_rd(double a, double b);
|
|
146
|
+
__device__ double __nv_ddiv_rn(double a, double b);
|
|
147
|
+
__device__ double __nv_ddiv_ru(double a, double b);
|
|
148
|
+
__device__ double __nv_ddiv_rz(double a, double b);
|
|
149
|
+
__device__ double __nv_dmul_rd(double a, double b);
|
|
150
|
+
__device__ double __nv_dmul_rn(double a, double b);
|
|
151
|
+
__device__ double __nv_dmul_ru(double a, double b);
|
|
152
|
+
__device__ double __nv_dmul_rz(double a, double b);
|
|
153
|
+
__device__ float __nv_double2float_rd(double a);
|
|
154
|
+
__device__ float __nv_double2float_rn(double a);
|
|
155
|
+
__device__ float __nv_double2float_ru(double a);
|
|
156
|
+
__device__ float __nv_double2float_rz(double a);
|
|
157
|
+
__device__ int __nv_double2hiint(double a);
|
|
158
|
+
__device__ int __nv_double2int_rd(double a);
|
|
159
|
+
__device__ int __nv_double2int_rn(double a);
|
|
160
|
+
__device__ int __nv_double2int_ru(double a);
|
|
161
|
+
__device__ int __nv_double2int_rz(double a);
|
|
162
|
+
__device__ long long __nv_double2ll_rd(double a);
|
|
163
|
+
__device__ long long __nv_double2ll_rn(double a);
|
|
164
|
+
__device__ long long __nv_double2ll_ru(double a);
|
|
165
|
+
__device__ long long __nv_double2ll_rz(double a);
|
|
166
|
+
__device__ int __nv_double2loint(double a);
|
|
167
|
+
__device__ unsigned int __nv_double2uint_rd(double a);
|
|
168
|
+
__device__ unsigned int __nv_double2uint_rn(double a);
|
|
169
|
+
__device__ unsigned int __nv_double2uint_ru(double a);
|
|
170
|
+
__device__ unsigned int __nv_double2uint_rz(double a);
|
|
171
|
+
__device__ unsigned long long __nv_double2ull_rd(double a);
|
|
172
|
+
__device__ unsigned long long __nv_double2ull_rn(double a);
|
|
173
|
+
__device__ unsigned long long __nv_double2ull_ru(double a);
|
|
174
|
+
__device__ unsigned long long __nv_double2ull_rz(double a);
|
|
175
|
+
__device__ unsigned long long __nv_double_as_longlong(double a);
|
|
176
|
+
__device__ double __nv_drcp_rd(double a);
|
|
177
|
+
__device__ double __nv_drcp_rn(double a);
|
|
178
|
+
__device__ double __nv_drcp_ru(double a);
|
|
179
|
+
__device__ double __nv_drcp_rz(double a);
|
|
180
|
+
__device__ double __nv_dsqrt_rd(double a);
|
|
181
|
+
__device__ double __nv_dsqrt_rn(double a);
|
|
182
|
+
__device__ double __nv_dsqrt_ru(double a);
|
|
183
|
+
__device__ double __nv_dsqrt_rz(double a);
|
|
184
|
+
__device__ double __nv_dsub_rd(double a, double b);
|
|
185
|
+
__device__ double __nv_dsub_rn(double a, double b);
|
|
186
|
+
__device__ double __nv_dsub_ru(double a, double b);
|
|
187
|
+
__device__ double __nv_dsub_rz(double a, double b);
|
|
188
|
+
__device__ double __nv_erfc(double a);
|
|
189
|
+
__device__ float __nv_erfcf(float a);
|
|
190
|
+
__device__ double __nv_erfcinv(double a);
|
|
191
|
+
__device__ float __nv_erfcinvf(float a);
|
|
192
|
+
__device__ double __nv_erfcx(double a);
|
|
193
|
+
__device__ float __nv_erfcxf(float a);
|
|
194
|
+
__device__ double __nv_erf(double a);
|
|
195
|
+
__device__ float __nv_erff(float a);
|
|
196
|
+
__device__ double __nv_erfinv(double a);
|
|
197
|
+
__device__ float __nv_erfinvf(float a);
|
|
198
|
+
__device__ double __nv_exp10(double a);
|
|
199
|
+
__device__ float __nv_exp10f(float a);
|
|
200
|
+
__device__ double __nv_exp2(double a);
|
|
201
|
+
__device__ float __nv_exp2f(float a);
|
|
202
|
+
__device__ double __nv_exp(double a);
|
|
203
|
+
__device__ float __nv_expf(float a);
|
|
204
|
+
__device__ double __nv_expm1(double a);
|
|
205
|
+
__device__ float __nv_expm1f(float a);
|
|
206
|
+
__device__ double __nv_fabs(double a);
|
|
207
|
+
__device__ float __nv_fabsf(float a);
|
|
208
|
+
__device__ float __nv_fadd_rd(float a, float b);
|
|
209
|
+
__device__ float __nv_fadd_rn(float a, float b);
|
|
210
|
+
__device__ float __nv_fadd_ru(float a, float b);
|
|
211
|
+
__device__ float __nv_fadd_rz(float a, float b);
|
|
212
|
+
__device__ float __nv_fast_cosf(float a);
|
|
213
|
+
__device__ float __nv_fast_exp10f(float a);
|
|
214
|
+
__device__ float __nv_fast_expf(float a);
|
|
215
|
+
__device__ float __nv_fast_fdividef(float a, float b);
|
|
216
|
+
__device__ float __nv_fast_log10f(float a);
|
|
217
|
+
__device__ float __nv_fast_log2f(float a);
|
|
218
|
+
__device__ float __nv_fast_logf(float a);
|
|
219
|
+
__device__ float __nv_fast_powf(float a, float b);
|
|
220
|
+
__device__ void __nv_fast_sincosf(float a, float *s, float *c);
|
|
221
|
+
__device__ float __nv_fast_sinf(float a);
|
|
222
|
+
__device__ float __nv_fast_tanf(float a);
|
|
223
|
+
__device__ double __nv_fdim(double a, double b);
|
|
224
|
+
__device__ float __nv_fdimf(float a, float b);
|
|
225
|
+
__device__ float __nv_fdiv_rd(float a, float b);
|
|
226
|
+
__device__ float __nv_fdiv_rn(float a, float b);
|
|
227
|
+
__device__ float __nv_fdiv_ru(float a, float b);
|
|
228
|
+
__device__ float __nv_fdiv_rz(float a, float b);
|
|
229
|
+
__device__ int __nv_ffs(int a);
|
|
230
|
+
__device__ int __nv_ffsll(long long a);
|
|
231
|
+
__device__ int __nv_finitef(float a);
|
|
232
|
+
__device__ unsigned short __nv_float2half_rn(float a);
|
|
233
|
+
__device__ int __nv_float2int_rd(float a);
|
|
234
|
+
__device__ int __nv_float2int_rn(float a);
|
|
235
|
+
__device__ int __nv_float2int_ru(float a);
|
|
236
|
+
__device__ int __nv_float2int_rz(float a);
|
|
237
|
+
__device__ long long __nv_float2ll_rd(float a);
|
|
238
|
+
__device__ long long __nv_float2ll_rn(float a);
|
|
239
|
+
__device__ long long __nv_float2ll_ru(float a);
|
|
240
|
+
__device__ long long __nv_float2ll_rz(float a);
|
|
241
|
+
__device__ unsigned int __nv_float2uint_rd(float a);
|
|
242
|
+
__device__ unsigned int __nv_float2uint_rn(float a);
|
|
243
|
+
__device__ unsigned int __nv_float2uint_ru(float a);
|
|
244
|
+
__device__ unsigned int __nv_float2uint_rz(float a);
|
|
245
|
+
__device__ unsigned long long __nv_float2ull_rd(float a);
|
|
246
|
+
__device__ unsigned long long __nv_float2ull_rn(float a);
|
|
247
|
+
__device__ unsigned long long __nv_float2ull_ru(float a);
|
|
248
|
+
__device__ unsigned long long __nv_float2ull_rz(float a);
|
|
249
|
+
__device__ int __nv_float_as_int(float a);
|
|
250
|
+
__device__ unsigned int __nv_float_as_uint(float a);
|
|
251
|
+
__device__ double __nv_floor(double a);
|
|
252
|
+
__device__ float __nv_floorf(float a);
|
|
253
|
+
__device__ double __nv_fma(double a, double b, double c);
|
|
254
|
+
__device__ float __nv_fmaf(float a, float b, float c);
|
|
255
|
+
__device__ float __nv_fmaf_ieee_rd(float a, float b, float c);
|
|
256
|
+
__device__ float __nv_fmaf_ieee_rn(float a, float b, float c);
|
|
257
|
+
__device__ float __nv_fmaf_ieee_ru(float a, float b, float c);
|
|
258
|
+
__device__ float __nv_fmaf_ieee_rz(float a, float b, float c);
|
|
259
|
+
__device__ float __nv_fmaf_rd(float a, float b, float c);
|
|
260
|
+
__device__ float __nv_fmaf_rn(float a, float b, float c);
|
|
261
|
+
__device__ float __nv_fmaf_ru(float a, float b, float c);
|
|
262
|
+
__device__ float __nv_fmaf_rz(float a, float b, float c);
|
|
263
|
+
__device__ double __nv_fma_rd(double a, double b, double c);
|
|
264
|
+
__device__ double __nv_fma_rn(double a, double b, double c);
|
|
265
|
+
__device__ double __nv_fma_ru(double a, double b, double c);
|
|
266
|
+
__device__ double __nv_fma_rz(double a, double b, double c);
|
|
267
|
+
__device__ double __nv_fmax(double a, double b);
|
|
268
|
+
__device__ float __nv_fmaxf(float a, float b);
|
|
269
|
+
__device__ double __nv_fmin(double a, double b);
|
|
270
|
+
__device__ float __nv_fminf(float a, float b);
|
|
271
|
+
__device__ double __nv_fmod(double a, double b);
|
|
272
|
+
__device__ float __nv_fmodf(float a, float b);
|
|
273
|
+
__device__ float __nv_fmul_rd(float a, float b);
|
|
274
|
+
__device__ float __nv_fmul_rn(float a, float b);
|
|
275
|
+
__device__ float __nv_fmul_ru(float a, float b);
|
|
276
|
+
__device__ float __nv_fmul_rz(float a, float b);
|
|
277
|
+
__device__ float __nv_frcp_rd(float a);
|
|
278
|
+
__device__ float __nv_frcp_rn(float a);
|
|
279
|
+
__device__ float __nv_frcp_ru(float a);
|
|
280
|
+
__device__ float __nv_frcp_rz(float a);
|
|
281
|
+
__device__ double __nv_frexp(double a, int *b);
|
|
282
|
+
__device__ float __nv_frexpf(float a, int *b);
|
|
283
|
+
__device__ float __nv_frsqrt_rn(float a);
|
|
284
|
+
__device__ float __nv_fsqrt_rd(float a);
|
|
285
|
+
__device__ float __nv_fsqrt_rn(float a);
|
|
286
|
+
__device__ float __nv_fsqrt_ru(float a);
|
|
287
|
+
__device__ float __nv_fsqrt_rz(float a);
|
|
288
|
+
__device__ float __nv_fsub_rd(float a, float b);
|
|
289
|
+
__device__ float __nv_fsub_rn(float a, float b);
|
|
290
|
+
__device__ float __nv_fsub_ru(float a, float b);
|
|
291
|
+
__device__ float __nv_fsub_rz(float a, float b);
|
|
292
|
+
__device__ int __nv_hadd(int a, int b);
|
|
293
|
+
__device__ float __nv_half2float(unsigned short h);
|
|
294
|
+
__device__ double __nv_hiloint2double(int a, int b);
|
|
295
|
+
__device__ double __nv_hypot(double a, double b);
|
|
296
|
+
__device__ float __nv_hypotf(float a, float b);
|
|
297
|
+
__device__ int __nv_ilogb(double a);
|
|
298
|
+
__device__ int __nv_ilogbf(float a);
|
|
299
|
+
__device__ double __nv_int2double_rn(int a);
|
|
300
|
+
__device__ float __nv_int2float_rd(int a);
|
|
301
|
+
__device__ float __nv_int2float_rn(int a);
|
|
302
|
+
__device__ float __nv_int2float_ru(int a);
|
|
303
|
+
__device__ float __nv_int2float_rz(int a);
|
|
304
|
+
__device__ float __nv_int_as_float(int a);
|
|
305
|
+
__device__ int __nv_isfinited(double a);
|
|
306
|
+
__device__ int __nv_isinfd(double a);
|
|
307
|
+
__device__ int __nv_isinff(float a);
|
|
308
|
+
__device__ int __nv_isnand(double a);
|
|
309
|
+
__device__ int __nv_isnanf(float a);
|
|
310
|
+
__device__ double __nv_j0(double a);
|
|
311
|
+
__device__ float __nv_j0f(float a);
|
|
312
|
+
__device__ double __nv_j1(double a);
|
|
313
|
+
__device__ float __nv_j1f(float a);
|
|
314
|
+
__device__ float __nv_jnf(int a, float b);
|
|
315
|
+
__device__ double __nv_jn(int a, double b);
|
|
316
|
+
__device__ double __nv_ldexp(double a, int b);
|
|
317
|
+
__device__ float __nv_ldexpf(float a, int b);
|
|
318
|
+
__device__ double __nv_lgamma(double a);
|
|
319
|
+
__device__ float __nv_lgammaf(float a);
|
|
320
|
+
__device__ double __nv_ll2double_rd(long long a);
|
|
321
|
+
__device__ double __nv_ll2double_rn(long long a);
|
|
322
|
+
__device__ double __nv_ll2double_ru(long long a);
|
|
323
|
+
__device__ double __nv_ll2double_rz(long long a);
|
|
324
|
+
__device__ float __nv_ll2float_rd(long long a);
|
|
325
|
+
__device__ float __nv_ll2float_rn(long long a);
|
|
326
|
+
__device__ float __nv_ll2float_ru(long long a);
|
|
327
|
+
__device__ float __nv_ll2float_rz(long long a);
|
|
328
|
+
__device__ long long __nv_llabs(long long a);
|
|
329
|
+
__device__ long long __nv_llmax(long long a, long long b);
|
|
330
|
+
__device__ long long __nv_llmin(long long a, long long b);
|
|
331
|
+
__device__ long long __nv_llrint(double a);
|
|
332
|
+
__device__ long long __nv_llrintf(float a);
|
|
333
|
+
__device__ long long __nv_llround(double a);
|
|
334
|
+
__device__ long long __nv_llroundf(float a);
|
|
335
|
+
__device__ double __nv_log10(double a);
|
|
336
|
+
__device__ float __nv_log10f(float a);
|
|
337
|
+
__device__ double __nv_log1p(double a);
|
|
338
|
+
__device__ float __nv_log1pf(float a);
|
|
339
|
+
__device__ double __nv_log2(double a);
|
|
340
|
+
__device__ float __nv_log2f(float a);
|
|
341
|
+
__device__ double __nv_logb(double a);
|
|
342
|
+
__device__ float __nv_logbf(float a);
|
|
343
|
+
__device__ double __nv_log(double a);
|
|
344
|
+
__device__ float __nv_logf(float a);
|
|
345
|
+
__device__ double __nv_longlong_as_double(long long a);
|
|
346
|
+
__device__ int __nv_max(int a, int b);
|
|
347
|
+
__device__ int __nv_min(int a, int b);
|
|
348
|
+
__device__ double __nv_modf(double a, double *b);
|
|
349
|
+
__device__ float __nv_modff(float a, float *b);
|
|
350
|
+
__device__ int __nv_mul24(int a, int b);
|
|
351
|
+
__device__ long long __nv_mul64hi(long long a, long long b);
|
|
352
|
+
__device__ int __nv_mulhi(int a, int b);
|
|
353
|
+
__device__ double __nv_nan(const signed char *a);
|
|
354
|
+
__device__ float __nv_nanf(const signed char *a);
|
|
355
|
+
__device__ double __nv_nearbyint(double a);
|
|
356
|
+
__device__ float __nv_nearbyintf(float a);
|
|
357
|
+
__device__ double __nv_nextafter(double a, double b);
|
|
358
|
+
__device__ float __nv_nextafterf(float a, float b);
|
|
359
|
+
__device__ double __nv_norm3d(double a, double b, double c);
|
|
360
|
+
__device__ float __nv_norm3df(float a, float b, float c);
|
|
361
|
+
__device__ double __nv_norm4d(double a, double b, double c, double d);
|
|
362
|
+
__device__ float __nv_norm4df(float a, float b, float c, float d);
|
|
363
|
+
__device__ double __nv_normcdf(double a);
|
|
364
|
+
__device__ float __nv_normcdff(float a);
|
|
365
|
+
__device__ double __nv_normcdfinv(double a);
|
|
366
|
+
__device__ float __nv_normcdfinvf(float a);
|
|
367
|
+
__device__ float __nv_normf(int a, const float *b);
|
|
368
|
+
__device__ double __nv_norm(int a, const double *b);
|
|
369
|
+
__device__ int __nv_popc(int a);
|
|
370
|
+
__device__ int __nv_popcll(long long a);
|
|
371
|
+
__device__ double __nv_pow(double a, double b);
|
|
372
|
+
__device__ float __nv_powf(float a, float b);
|
|
373
|
+
__device__ double __nv_powi(double a, int b);
|
|
374
|
+
__device__ float __nv_powif(float a, int b);
|
|
375
|
+
__device__ double __nv_rcbrt(double a);
|
|
376
|
+
__device__ float __nv_rcbrtf(float a);
|
|
377
|
+
__device__ double __nv_rcp64h(double a);
|
|
378
|
+
__device__ double __nv_remainder(double a, double b);
|
|
379
|
+
__device__ float __nv_remainderf(float a, float b);
|
|
380
|
+
__device__ double __nv_remquo(double a, double b, int *c);
|
|
381
|
+
__device__ float __nv_remquof(float a, float b, int *c);
|
|
382
|
+
__device__ int __nv_rhadd(int a, int b);
|
|
383
|
+
__device__ double __nv_rhypot(double a, double b);
|
|
384
|
+
__device__ float __nv_rhypotf(float a, float b);
|
|
385
|
+
__device__ double __nv_rint(double a);
|
|
386
|
+
__device__ float __nv_rintf(float a);
|
|
387
|
+
__device__ double __nv_rnorm3d(double a, double b, double c);
|
|
388
|
+
__device__ float __nv_rnorm3df(float a, float b, float c);
|
|
389
|
+
__device__ double __nv_rnorm4d(double a, double b, double c, double d);
|
|
390
|
+
__device__ float __nv_rnorm4df(float a, float b, float c, float d);
|
|
391
|
+
__device__ float __nv_rnormf(int a, const float *b);
|
|
392
|
+
__device__ double __nv_rnorm(int a, const double *b);
|
|
393
|
+
__device__ double __nv_round(double a);
|
|
394
|
+
__device__ float __nv_roundf(float a);
|
|
395
|
+
__device__ double __nv_rsqrt(double a);
|
|
396
|
+
__device__ float __nv_rsqrtf(float a);
|
|
397
|
+
__device__ int __nv_sad(int a, int b, int c);
|
|
398
|
+
__device__ float __nv_saturatef(float a);
|
|
399
|
+
__device__ double __nv_scalbn(double a, int b);
|
|
400
|
+
__device__ float __nv_scalbnf(float a, int b);
|
|
401
|
+
__device__ int __nv_signbitd(double a);
|
|
402
|
+
__device__ int __nv_signbitf(float a);
|
|
403
|
+
__device__ void __nv_sincos(double a, double *b, double *c);
|
|
404
|
+
__device__ void __nv_sincosf(float a, float *b, float *c);
|
|
405
|
+
__device__ void __nv_sincospi(double a, double *b, double *c);
|
|
406
|
+
__device__ void __nv_sincospif(float a, float *b, float *c);
|
|
407
|
+
__device__ double __nv_sin(double a);
|
|
408
|
+
__device__ float __nv_sinf(float a);
|
|
409
|
+
__device__ double __nv_sinh(double a);
|
|
410
|
+
__device__ float __nv_sinhf(float a);
|
|
411
|
+
__device__ double __nv_sinpi(double a);
|
|
412
|
+
__device__ float __nv_sinpif(float a);
|
|
413
|
+
__device__ double __nv_sqrt(double a);
|
|
414
|
+
__device__ float __nv_sqrtf(float a);
|
|
415
|
+
__device__ double __nv_tan(double a);
|
|
416
|
+
__device__ float __nv_tanf(float a);
|
|
417
|
+
__device__ double __nv_tanh(double a);
|
|
418
|
+
__device__ float __nv_tanhf(float a);
|
|
419
|
+
__device__ double __nv_tgamma(double a);
|
|
420
|
+
__device__ float __nv_tgammaf(float a);
|
|
421
|
+
__device__ double __nv_trunc(double a);
|
|
422
|
+
__device__ float __nv_truncf(float a);
|
|
423
|
+
__device__ int __nv_uhadd(unsigned int a, unsigned int b);
|
|
424
|
+
__device__ double __nv_uint2double_rn(unsigned int i);
|
|
425
|
+
__device__ float __nv_uint2float_rd(unsigned int a);
|
|
426
|
+
__device__ float __nv_uint2float_rn(unsigned int a);
|
|
427
|
+
__device__ float __nv_uint2float_ru(unsigned int a);
|
|
428
|
+
__device__ float __nv_uint2float_rz(unsigned int a);
|
|
429
|
+
__device__ float __nv_uint_as_float(unsigned int a);
|
|
430
|
+
__device__ double __nv_ull2double_rd(unsigned long long a);
|
|
431
|
+
__device__ double __nv_ull2double_rn(unsigned long long a);
|
|
432
|
+
__device__ double __nv_ull2double_ru(unsigned long long a);
|
|
433
|
+
__device__ double __nv_ull2double_rz(unsigned long long a);
|
|
434
|
+
__device__ float __nv_ull2float_rd(unsigned long long a);
|
|
435
|
+
__device__ float __nv_ull2float_rn(unsigned long long a);
|
|
436
|
+
__device__ float __nv_ull2float_ru(unsigned long long a);
|
|
437
|
+
__device__ float __nv_ull2float_rz(unsigned long long a);
|
|
438
|
+
__device__ unsigned long long __nv_ullmax(unsigned long long a, unsigned long long b);
|
|
439
|
+
__device__ unsigned long long __nv_ullmin(unsigned long long a, unsigned long long b);
|
|
440
|
+
__device__ unsigned int __nv_umax(unsigned int a, unsigned int b);
|
|
441
|
+
__device__ unsigned int __nv_umin(unsigned int a, unsigned int b);
|
|
442
|
+
__device__ unsigned int __nv_umul24(unsigned int a, unsigned int b);
|
|
443
|
+
__device__ unsigned long long __nv_umul64hi(unsigned long long a, unsigned long long b);
|
|
444
|
+
__device__ unsigned int __nv_umulhi(unsigned int a, unsigned int b);
|
|
445
|
+
__device__ unsigned int __nv_urhadd(unsigned int a, unsigned int b);
|
|
446
|
+
__device__ unsigned int __nv_usad(unsigned int a, unsigned int b, unsigned int c);
|
|
447
|
+
__device__ double __nv_y0(double a);
|
|
448
|
+
__device__ float __nv_y0f(float a);
|
|
449
|
+
__device__ double __nv_y1(double a);
|
|
450
|
+
__device__ float __nv_y1f(float a);
|
|
451
|
+
__device__ float __nv_ynf(int a, float b);
|
|
452
|
+
__device__ double __nv_yn(int a, double b);
|
|
453
|
+
|
|
454
|
+
} // extern "C"
|
|
455
|
+
|
|
456
|
+
// Implementation of CUDA intrinsics
|
|
457
|
+
__device_forceinline__ int __all(int a) { return __nvvm_vote_all(a); }
|
|
458
|
+
__device_forceinline__ int __any(int a) { return __nvvm_vote_any(a); }
|
|
459
|
+
__device_forceinline__ unsigned int __ballot(int a) { return __nvvm_vote_ballot(a); }
|
|
460
|
+
__device_forceinline__ unsigned int __brev(unsigned int a) { return __nv_brev(a); }
|
|
461
|
+
__device_forceinline__ unsigned long long __brevll(unsigned long long a) { return __nv_brevll(a); }
|
|
462
|
+
__device_forceinline__ void __brkpt() { __asm__ __volatile__("brkpt;"); }
|
|
463
|
+
__device_forceinline__ void __brkpt(int a) { __brkpt(); }
|
|
464
|
+
__device_forceinline__ unsigned int __byte_perm(unsigned int a, unsigned int b, unsigned int c) { return __nv_byte_perm(a, b, c); }
|
|
465
|
+
__device_forceinline__ int __clz(int a) { return __nv_clz(a); }
|
|
466
|
+
__device_forceinline__ int __clzll(long long a) { return __nv_clzll(a); }
|
|
467
|
+
__device_forceinline__ float __cosf(float a) { return __nv_fast_cosf(a); }
|
|
468
|
+
__device_forceinline__ double __dAtomicAdd(double *p, double v) { return __nvvm_atom_add_gen_d(p, v); }
|
|
469
|
+
__device_forceinline__ double __dAtomicAdd_block(double *p, double v) { return __nvvm_atom_cta_add_gen_d(p, v); }
|
|
470
|
+
__device_forceinline__ double __dAtomicAdd_system(double *p, double v) { return __nvvm_atom_sys_add_gen_d(p, v); }
|
|
471
|
+
__device_forceinline__ double __dadd_rd(double a, double b) { return __nv_dadd_rd(a, b); }
|
|
472
|
+
__device_forceinline__ double __dadd_rn(double a, double b) { return __nv_dadd_rn(a, b); }
|
|
473
|
+
__device_forceinline__ double __dadd_ru(double a, double b) { return __nv_dadd_ru(a, b); }
|
|
474
|
+
__device_forceinline__ double __dadd_rz(double a, double b) { return __nv_dadd_rz(a, b); }
|
|
475
|
+
__device_forceinline__ double __ddiv_rd(double a, double b) { return __nv_ddiv_rd(a, b); }
|
|
476
|
+
__device_forceinline__ double __ddiv_rn(double a, double b) { return __nv_ddiv_rn(a, b); }
|
|
477
|
+
__device_forceinline__ double __ddiv_ru(double a, double b) { return __nv_ddiv_ru(a, b); }
|
|
478
|
+
__device_forceinline__ double __ddiv_rz(double a, double b) { return __nv_ddiv_rz(a, b); }
|
|
479
|
+
__device_forceinline__ double __dmul_rd(double a, double b) { return __nv_dmul_rd(a, b); }
|
|
480
|
+
__device_forceinline__ double __dmul_rn(double a, double b) { return __nv_dmul_rn(a, b); }
|
|
481
|
+
__device_forceinline__ double __dmul_ru(double a, double b) { return __nv_dmul_ru(a, b); }
|
|
482
|
+
__device_forceinline__ double __dmul_rz(double a, double b) { return __nv_dmul_rz(a, b); }
|
|
483
|
+
__device_forceinline__ float __double2float_rd(double a) { return __nv_double2float_rd(a); }
|
|
484
|
+
__device_forceinline__ float __double2float_rn(double a) { return __nv_double2float_rn(a); }
|
|
485
|
+
__device_forceinline__ float __double2float_ru(double a) { return __nv_double2float_ru(a); }
|
|
486
|
+
__device_forceinline__ float __double2float_rz(double a) { return __nv_double2float_rz(a); }
|
|
487
|
+
__device_forceinline__ int __double2hiint(double a) { return __nv_double2hiint(a); }
|
|
488
|
+
__device_forceinline__ int __double2int_rd(double a) { return __nv_double2int_rd(a); }
|
|
489
|
+
__device_forceinline__ int __double2int_rn(double a) { return __nv_double2int_rn(a); }
|
|
490
|
+
__device_forceinline__ int __double2int_ru(double a) { return __nv_double2int_ru(a); }
|
|
491
|
+
__device_forceinline__ int __double2int_rz(double a) { return __nv_double2int_rz(a); }
|
|
492
|
+
__device_forceinline__ long long __double2ll_rd(double a) { return __nv_double2ll_rd(a); }
|
|
493
|
+
__device_forceinline__ long long __double2ll_rn(double a) { return __nv_double2ll_rn(a); }
|
|
494
|
+
__device_forceinline__ long long __double2ll_ru(double a) { return __nv_double2ll_ru(a); }
|
|
495
|
+
__device_forceinline__ long long __double2ll_rz(double a) { return __nv_double2ll_rz(a); }
|
|
496
|
+
__device_forceinline__ int __double2loint(double a) { return __nv_double2loint(a); }
|
|
497
|
+
__device_forceinline__ unsigned int __double2uint_rd(double a) { return __nv_double2uint_rd(a); }
|
|
498
|
+
__device_forceinline__ unsigned int __double2uint_rn(double a) { return __nv_double2uint_rn(a); }
|
|
499
|
+
__device_forceinline__ unsigned int __double2uint_ru(double a) { return __nv_double2uint_ru(a); }
|
|
500
|
+
__device_forceinline__ unsigned int __double2uint_rz(double a) { return __nv_double2uint_rz(a); }
|
|
501
|
+
__device_forceinline__ unsigned long long __double2ull_rd(double a) { return __nv_double2ull_rd(a); }
|
|
502
|
+
__device_forceinline__ unsigned long long __double2ull_rn(double a) { return __nv_double2ull_rn(a); }
|
|
503
|
+
__device_forceinline__ unsigned long long __double2ull_ru(double a) { return __nv_double2ull_ru(a); }
|
|
504
|
+
__device_forceinline__ unsigned long long __double2ull_rz(double a) { return __nv_double2ull_rz(a); }
|
|
505
|
+
__device_forceinline__ long long __double_as_longlong(double a) { return __nv_double_as_longlong(a); }
|
|
506
|
+
__device_forceinline__ double __drcp_rd(double a) { return __nv_drcp_rd(a); }
|
|
507
|
+
__device_forceinline__ double __drcp_rn(double a) { return __nv_drcp_rn(a); }
|
|
508
|
+
__device_forceinline__ double __drcp_ru(double a) { return __nv_drcp_ru(a); }
|
|
509
|
+
__device_forceinline__ double __drcp_rz(double a) { return __nv_drcp_rz(a); }
|
|
510
|
+
__device_forceinline__ double __dsqrt_rd(double a) { return __nv_dsqrt_rd(a); }
|
|
511
|
+
__device_forceinline__ double __dsqrt_rn(double a) { return __nv_dsqrt_rn(a); }
|
|
512
|
+
__device_forceinline__ double __dsqrt_ru(double a) { return __nv_dsqrt_ru(a); }
|
|
513
|
+
__device_forceinline__ double __dsqrt_rz(double a) { return __nv_dsqrt_rz(a); }
|
|
514
|
+
__device_forceinline__ double __dsub_rd(double a, double b) { return __nv_dsub_rd(a, b); }
|
|
515
|
+
__device_forceinline__ double __dsub_rn(double a, double b) { return __nv_dsub_rn(a, b); }
|
|
516
|
+
__device_forceinline__ double __dsub_ru(double a, double b) { return __nv_dsub_ru(a, b); }
|
|
517
|
+
__device_forceinline__ double __dsub_rz(double a, double b) { return __nv_dsub_rz(a, b); }
|
|
518
|
+
__device_forceinline__ float __exp10f(float a) { return __nv_fast_exp10f(a); }
|
|
519
|
+
__device_forceinline__ float __expf(float a) { return __nv_fast_expf(a); }
|
|
520
|
+
__device_forceinline__ float __fAtomicAdd(float *p, float v) { return __nvvm_atom_add_gen_f(p, v); }
|
|
521
|
+
__device_forceinline__ float __fAtomicAdd_block(float *p, float v) { return __nvvm_atom_cta_add_gen_f(p, v); }
|
|
522
|
+
__device_forceinline__ float __fAtomicAdd_system(float *p, float v) { return __nvvm_atom_sys_add_gen_f(p, v); }
|
|
523
|
+
__device_forceinline__ float __fAtomicExch(float *p, float v) { return __nv_int_as_float(__nvvm_atom_xchg_gen_i((int *)p, __nv_float_as_int(v))); }
|
|
524
|
+
__device_forceinline__ float __fAtomicExch_block(float *p, float v) { return __nv_int_as_float(__nvvm_atom_cta_xchg_gen_i((int *)p, __nv_float_as_int(v))); }
|
|
525
|
+
__device_forceinline__ float __fAtomicExch_system(float *p, float v) { return __nv_int_as_float(__nvvm_atom_sys_xchg_gen_i((int *)p, __nv_float_as_int(v))); }
|
|
526
|
+
__device_forceinline__ float __fadd_rd(float a, float b) { return __nv_fadd_rd(a, b); }
|
|
527
|
+
__device_forceinline__ float __fadd_rn(float a, float b) { return __nv_fadd_rn(a, b); }
|
|
528
|
+
__device_forceinline__ float __fadd_ru(float a, float b) { return __nv_fadd_ru(a, b); }
|
|
529
|
+
__device_forceinline__ float __fadd_rz(float a, float b) { return __nv_fadd_rz(a, b); }
|
|
530
|
+
__device_forceinline__ float __fdiv_rd(float a, float b) { return __nv_fdiv_rd(a, b); }
|
|
531
|
+
__device_forceinline__ float __fdiv_rn(float a, float b) { return __nv_fdiv_rn(a, b); }
|
|
532
|
+
__device_forceinline__ float __fdiv_ru(float a, float b) { return __nv_fdiv_ru(a, b); }
|
|
533
|
+
__device_forceinline__ float __fdiv_rz(float a, float b) { return __nv_fdiv_rz(a, b); }
|
|
534
|
+
__device_forceinline__ float __fdividef(float a, float b) { return __nv_fast_fdividef(a, b); }
|
|
535
|
+
__device_forceinline__ int __ffs(int a) { return __nv_ffs(a); }
|
|
536
|
+
__device_forceinline__ int __ffsll(long long a) { return __nv_ffsll(a); }
|
|
537
|
+
__device_forceinline__ int __finite(double a) { return __nv_isfinited(a); }
|
|
538
|
+
__device_forceinline__ int __finitef(float a) { return __nv_finitef(a); }
|
|
539
|
+
__device_forceinline__ int __float2int_rd(float a) { return __nv_float2int_rd(a); }
|
|
540
|
+
__device_forceinline__ int __float2int_rn(float a) { return __nv_float2int_rn(a); }
|
|
541
|
+
__device_forceinline__ int __float2int_ru(float a) { return __nv_float2int_ru(a); }
|
|
542
|
+
__device_forceinline__ int __float2int_rz(float a) { return __nv_float2int_rz(a); }
|
|
543
|
+
__device_forceinline__ long long __float2ll_rd(float a) { return __nv_float2ll_rd(a); }
|
|
544
|
+
__device_forceinline__ long long __float2ll_rn(float a) { return __nv_float2ll_rn(a); }
|
|
545
|
+
__device_forceinline__ long long __float2ll_ru(float a) { return __nv_float2ll_ru(a); }
|
|
546
|
+
__device_forceinline__ long long __float2ll_rz(float a) { return __nv_float2ll_rz(a); }
|
|
547
|
+
__device_forceinline__ unsigned int __float2uint_rd(float a) { return __nv_float2uint_rd(a); }
|
|
548
|
+
__device_forceinline__ unsigned int __float2uint_rn(float a) { return __nv_float2uint_rn(a); }
|
|
549
|
+
__device_forceinline__ unsigned int __float2uint_ru(float a) { return __nv_float2uint_ru(a); }
|
|
550
|
+
__device_forceinline__ unsigned int __float2uint_rz(float a) { return __nv_float2uint_rz(a); }
|
|
551
|
+
__device_forceinline__ unsigned long long __float2ull_rd(float a) { return __nv_float2ull_rd(a); }
|
|
552
|
+
__device_forceinline__ unsigned long long __float2ull_rn(float a) { return __nv_float2ull_rn(a); }
|
|
553
|
+
__device_forceinline__ unsigned long long __float2ull_ru(float a) { return __nv_float2ull_ru(a); }
|
|
554
|
+
__device_forceinline__ unsigned long long __float2ull_rz(float a) { return __nv_float2ull_rz(a); }
|
|
555
|
+
__device_forceinline__ int __float_as_int(float a) { return __nv_float_as_int(a); }
|
|
556
|
+
__device_forceinline__ unsigned int __float_as_uint(float a) { return __nv_float_as_uint(a); }
|
|
557
|
+
__device_forceinline__ double __fma_rd(double a, double b, double c) { return __nv_fma_rd(a, b, c); }
|
|
558
|
+
__device_forceinline__ double __fma_rn(double a, double b, double c) { return __nv_fma_rn(a, b, c); }
|
|
559
|
+
__device_forceinline__ double __fma_ru(double a, double b, double c) { return __nv_fma_ru(a, b, c); }
|
|
560
|
+
__device_forceinline__ double __fma_rz(double a, double b, double c) { return __nv_fma_rz(a, b, c); }
|
|
561
|
+
__device_forceinline__ float __fmaf_ieee_rd(float a, float b, float c) { return __nv_fmaf_ieee_rd(a, b, c); }
|
|
562
|
+
__device_forceinline__ float __fmaf_ieee_rn(float a, float b, float c) { return __nv_fmaf_ieee_rn(a, b, c); }
|
|
563
|
+
__device_forceinline__ float __fmaf_ieee_ru(float a, float b, float c) { return __nv_fmaf_ieee_ru(a, b, c); }
|
|
564
|
+
__device_forceinline__ float __fmaf_ieee_rz(float a, float b, float c) { return __nv_fmaf_ieee_rz(a, b, c); }
|
|
565
|
+
__device_forceinline__ float __fmaf_rd(float a, float b, float c) { return __nv_fmaf_rd(a, b, c); }
|
|
566
|
+
__device_forceinline__ float __fmaf_rn(float a, float b, float c) { return __nv_fmaf_rn(a, b, c); }
|
|
567
|
+
__device_forceinline__ float __fmaf_ru(float a, float b, float c) { return __nv_fmaf_ru(a, b, c); }
|
|
568
|
+
__device_forceinline__ float __fmaf_rz(float a, float b, float c) { return __nv_fmaf_rz(a, b, c); }
|
|
569
|
+
__device_forceinline__ float __fmul_rd(float a, float b) { return __nv_fmul_rd(a, b); }
|
|
570
|
+
__device_forceinline__ float __fmul_rn(float a, float b) { return __nv_fmul_rn(a, b); }
|
|
571
|
+
__device_forceinline__ float __fmul_ru(float a, float b) { return __nv_fmul_ru(a, b); }
|
|
572
|
+
__device_forceinline__ float __fmul_rz(float a, float b) { return __nv_fmul_rz(a, b); }
|
|
573
|
+
__device_forceinline__ float __frcp_rd(float a) { return __nv_frcp_rd(a); }
|
|
574
|
+
__device_forceinline__ float __frcp_rn(float a) { return __nv_frcp_rn(a); }
|
|
575
|
+
__device_forceinline__ float __frcp_ru(float a) { return __nv_frcp_ru(a); }
|
|
576
|
+
__device_forceinline__ float __frcp_rz(float a) { return __nv_frcp_rz(a); }
|
|
577
|
+
__device_forceinline__ float __frsqrt_rn(float a) { return __nv_frsqrt_rn(a); }
|
|
578
|
+
__device_forceinline__ float __fsqrt_rd(float a) { return __nv_fsqrt_rd(a); }
|
|
579
|
+
__device_forceinline__ float __fsqrt_rn(float a) { return __nv_fsqrt_rn(a); }
|
|
580
|
+
__device_forceinline__ float __fsqrt_ru(float a) { return __nv_fsqrt_ru(a); }
|
|
581
|
+
__device_forceinline__ float __fsqrt_rz(float a) { return __nv_fsqrt_rz(a); }
|
|
582
|
+
__device_forceinline__ float __fsub_rd(float a, float b) { return __nv_fsub_rd(a, b); }
|
|
583
|
+
__device_forceinline__ float __fsub_rn(float a, float b) { return __nv_fsub_rn(a, b); }
|
|
584
|
+
__device_forceinline__ float __fsub_ru(float a, float b) { return __nv_fsub_ru(a, b); }
|
|
585
|
+
__device_forceinline__ float __fsub_rz(float a, float b) { return __nv_fsub_rz(a, b); }
|
|
586
|
+
__device_forceinline__ int __hadd(int a, int b) { return __nv_hadd(a, b); }
|
|
587
|
+
__device_forceinline__ double __hiloint2double(int a, int b) { return __nv_hiloint2double(a, b); }
|
|
588
|
+
__device_forceinline__ int __iAtomicAdd(int *p, int v) { return __nvvm_atom_add_gen_i(p, v); }
|
|
589
|
+
__device_forceinline__ int __iAtomicAdd_block(int *p, int v) { return __nvvm_atom_cta_add_gen_i(p, v); }
|
|
590
|
+
__device_forceinline__ int __iAtomicAdd_system(int *p, int v) { return __nvvm_atom_sys_add_gen_i(p, v); }
|
|
591
|
+
__device_forceinline__ int __iAtomicAnd(int *p, int v) { return __nvvm_atom_and_gen_i(p, v); }
|
|
592
|
+
__device_forceinline__ int __iAtomicAnd_block(int *p, int v) { return __nvvm_atom_cta_and_gen_i(p, v); }
|
|
593
|
+
__device_forceinline__ int __iAtomicAnd_system(int *p, int v) { return __nvvm_atom_sys_and_gen_i(p, v); }
|
|
594
|
+
__device_forceinline__ int __iAtomicCAS(int *p, int cmp, int v) { return __nvvm_atom_cas_gen_i(p, cmp, v); }
|
|
595
|
+
__device_forceinline__ int __iAtomicCAS_block(int *p, int cmp, int v) { return __nvvm_atom_cta_cas_gen_i(p, cmp, v); }
|
|
596
|
+
__device_forceinline__ int __iAtomicCAS_system(int *p, int cmp, int v) { return __nvvm_atom_sys_cas_gen_i(p, cmp, v); }
|
|
597
|
+
__device_forceinline__ int __iAtomicExch(int *p, int v) { return __nvvm_atom_xchg_gen_i(p, v); }
|
|
598
|
+
__device_forceinline__ int __iAtomicExch_block(int *p, int v) { return __nvvm_atom_cta_xchg_gen_i(p, v); }
|
|
599
|
+
__device_forceinline__ int __iAtomicExch_system(int *p, int v) { return __nvvm_atom_sys_xchg_gen_i(p, v); }
|
|
600
|
+
__device_forceinline__ int __iAtomicMax(int *p, int v) { return __nvvm_atom_max_gen_i(p, v); }
|
|
601
|
+
__device_forceinline__ int __iAtomicMax_block(int *p, int v) { return __nvvm_atom_cta_max_gen_i(p, v); }
|
|
602
|
+
__device_forceinline__ int __iAtomicMax_system(int *p, int v) { return __nvvm_atom_sys_max_gen_i(p, v); }
|
|
603
|
+
__device_forceinline__ int __iAtomicMin(int *p, int v) { return __nvvm_atom_min_gen_i(p, v); }
|
|
604
|
+
__device_forceinline__ int __iAtomicMin_block(int *p, int v) { return __nvvm_atom_cta_min_gen_i(p, v); }
|
|
605
|
+
__device_forceinline__ int __iAtomicMin_system(int *p, int v) { return __nvvm_atom_sys_min_gen_i(p, v); }
|
|
606
|
+
__device_forceinline__ int __iAtomicOr(int *p, int v) { return __nvvm_atom_or_gen_i(p, v); }
|
|
607
|
+
__device_forceinline__ int __iAtomicOr_block(int *p, int v) { return __nvvm_atom_cta_or_gen_i(p, v); }
|
|
608
|
+
__device_forceinline__ int __iAtomicOr_system(int *p, int v) { return __nvvm_atom_sys_or_gen_i(p, v); }
|
|
609
|
+
__device_forceinline__ int __iAtomicXor(int *p, int v) { return __nvvm_atom_xor_gen_i(p, v); }
|
|
610
|
+
__device_forceinline__ int __iAtomicXor_block(int *p, int v) { return __nvvm_atom_cta_xor_gen_i(p, v); }
|
|
611
|
+
__device_forceinline__ int __iAtomicXor_system(int *p, int v) { return __nvvm_atom_sys_xor_gen_i(p, v); }
|
|
612
|
+
__device_forceinline__ long long __illAtomicMax(long long *p, long long v) { return __nvvm_atom_max_gen_ll(p, v); }
|
|
613
|
+
__device_forceinline__ long long __illAtomicMax_block(long long *p, long long v) { return __nvvm_atom_cta_max_gen_ll(p, v); }
|
|
614
|
+
__device_forceinline__ long long __illAtomicMax_system(long long *p, long long v) { return __nvvm_atom_sys_max_gen_ll(p, v); }
|
|
615
|
+
__device_forceinline__ long long __illAtomicMin(long long *p, long long v) { return __nvvm_atom_min_gen_ll(p, v); }
|
|
616
|
+
__device_forceinline__ long long __illAtomicMin_block(long long *p, long long v) { return __nvvm_atom_cta_min_gen_ll(p, v); }
|
|
617
|
+
__device_forceinline__ long long __illAtomicMin_system(long long *p, long long v) { return __nvvm_atom_sys_min_gen_ll(p, v); }
|
|
618
|
+
__device_forceinline__ double __int2double_rn(int a) { return __nv_int2double_rn(a); }
|
|
619
|
+
__device_forceinline__ float __int2float_rd(int a) { return __nv_int2float_rd(a); }
|
|
620
|
+
__device_forceinline__ float __int2float_rn(int a) { return __nv_int2float_rn(a); }
|
|
621
|
+
__device_forceinline__ float __int2float_ru(int a) { return __nv_int2float_ru(a); }
|
|
622
|
+
__device_forceinline__ float __int2float_rz(int a) { return __nv_int2float_rz(a); }
|
|
623
|
+
__device_forceinline__ float __int_as_float(int a) { return __nv_int_as_float(a); }
|
|
624
|
+
__device_forceinline__ int __isfinited(double a) { return __nv_isfinited(a); }
|
|
625
|
+
__device_forceinline__ int __isinf(double a) { return __nv_isinfd(a); }
|
|
626
|
+
__device_forceinline__ int __isinff(float a) { return __nv_isinff(a); }
|
|
627
|
+
__device_forceinline__ int __isnan(double a) { return __nv_isnand(a); }
|
|
628
|
+
__device_forceinline__ int __isnanf(float a) { return __nv_isnanf(a); }
|
|
629
|
+
__device_forceinline__ double __ll2double_rd(long long a) { return __nv_ll2double_rd(a); }
|
|
630
|
+
__device_forceinline__ double __ll2double_rn(long long a) { return __nv_ll2double_rn(a); }
|
|
631
|
+
__device_forceinline__ double __ll2double_ru(long long a) { return __nv_ll2double_ru(a); }
|
|
632
|
+
__device_forceinline__ double __ll2double_rz(long long a) { return __nv_ll2double_rz(a); }
|
|
633
|
+
__device_forceinline__ float __ll2float_rd(long long a) { return __nv_ll2float_rd(a); }
|
|
634
|
+
__device_forceinline__ float __ll2float_rn(long long a) { return __nv_ll2float_rn(a); }
|
|
635
|
+
__device_forceinline__ float __ll2float_ru(long long a) { return __nv_ll2float_ru(a); }
|
|
636
|
+
__device_forceinline__ float __ll2float_rz(long long a) { return __nv_ll2float_rz(a); }
|
|
637
|
+
__device_forceinline__ long long __llAtomicAnd(long long *p, long long v) { return __nvvm_atom_and_gen_ll(p, v); }
|
|
638
|
+
__device_forceinline__ long long __llAtomicAnd_block(long long *p, long long v) { return __nvvm_atom_cta_and_gen_ll(p, v); }
|
|
639
|
+
__device_forceinline__ long long __llAtomicAnd_system(long long *p, long long v) { return __nvvm_atom_sys_and_gen_ll(p, v); }
|
|
640
|
+
__device_forceinline__ long long __llAtomicOr(long long *p, long long v) { return __nvvm_atom_or_gen_ll(p, v); }
|
|
641
|
+
__device_forceinline__ long long __llAtomicOr_block(long long *p, long long v) { return __nvvm_atom_cta_or_gen_ll(p, v); }
|
|
642
|
+
__device_forceinline__ long long __llAtomicOr_system(long long *p, long long v) { return __nvvm_atom_sys_or_gen_ll(p, v); }
|
|
643
|
+
__device_forceinline__ long long __llAtomicXor(long long *p, long long v) { return __nvvm_atom_xor_gen_ll(p, v); }
|
|
644
|
+
__device_forceinline__ long long __llAtomicXor_block(long long *p, long long v) { return __nvvm_atom_cta_xor_gen_ll(p, v); }
|
|
645
|
+
__device_forceinline__ long long __llAtomicXor_system(long long *p, long long v) { return __nvvm_atom_sys_xor_gen_ll(p, v); }
|
|
646
|
+
__device_forceinline__ float __log10f(float a) { return __nv_fast_log10f(a); }
|
|
647
|
+
__device_forceinline__ float __log2f(float a) { return __nv_fast_log2f(a); }
|
|
648
|
+
__device_forceinline__ float __logf(float a) { return __nv_fast_logf(a); }
|
|
649
|
+
__device_forceinline__ double __longlong_as_double(long long a) { return __nv_longlong_as_double(a); }
|
|
650
|
+
__device_forceinline__ int __mul24(int a, int b) { return __nv_mul24(a, b); }
|
|
651
|
+
__device_forceinline__ long long __mul64hi(long long a, long long b) { return __nv_mul64hi(a, b); }
|
|
652
|
+
__device_forceinline__ int __mulhi(int a, int b) { return __nv_mulhi(a, b); }
|
|
653
|
+
__device_forceinline__ unsigned int __pm0(void) { return __nvvm_read_ptx_sreg_pm0(); }
|
|
654
|
+
__device_forceinline__ unsigned int __pm1(void) { return __nvvm_read_ptx_sreg_pm1(); }
|
|
655
|
+
__device_forceinline__ unsigned int __pm2(void) { return __nvvm_read_ptx_sreg_pm2(); }
|
|
656
|
+
__device_forceinline__ unsigned int __pm3(void) { return __nvvm_read_ptx_sreg_pm3(); }
|
|
657
|
+
__device_forceinline__ int __popc(int a) { return __nv_popc(a); }
|
|
658
|
+
__device_forceinline__ int __popcll(long long a) { return __nv_popcll(a); }
|
|
659
|
+
__device_forceinline__ float __powf(float a, float b) { return __nv_fast_powf(a, b); }
|
|
660
|
+
#define __prof_trigger(__counter) __asm__ __volatile__("pmevent \t%0;" ::"i"(__counter))
|
|
661
|
+
__device_forceinline__ int __rhadd(int a, int b) { return __nv_rhadd(a, b); }
|
|
662
|
+
__device_forceinline__ unsigned int __sad(int a, int b, unsigned int c) { return __nv_sad(a, b, c); }
|
|
663
|
+
__device_forceinline__ float __saturatef(float a) { return __nv_saturatef(a); }
|
|
664
|
+
__device_forceinline__ int __signbitd(double a) { return __nv_signbitd(a); }
|
|
665
|
+
__device_forceinline__ int __signbitf(float a) { return __nv_signbitf(a); }
|
|
666
|
+
__device_forceinline__ void __sincosf(float a, float *s, float *c) { return __nv_fast_sincosf(a, s, c); }
|
|
667
|
+
__device_forceinline__ float __sinf(float a) { return __nv_fast_sinf(a); }
|
|
668
|
+
__device_forceinline__ int __syncthreads_and(int a) { return __nvvm_bar0_and(a); }
|
|
669
|
+
__device_forceinline__ int __syncthreads_count(int a) { return __nvvm_bar0_popc(a); }
|
|
670
|
+
__device_forceinline__ int __syncthreads_or(int a) { return __nvvm_bar0_or(a); }
|
|
671
|
+
__device_forceinline__ float __tanf(float a) { return __nv_fast_tanf(a); }
|
|
672
|
+
__device_forceinline__ void __threadfence(void) { __nvvm_membar_gl(); }
|
|
673
|
+
__device_forceinline__ void __threadfence_block(void) { __nvvm_membar_cta(); };
|
|
674
|
+
__device_forceinline__ void __threadfence_system(void) { __nvvm_membar_sys(); };
|
|
675
|
+
__device_forceinline__ void __trap(void) { __asm__ __volatile__("trap;"); }
|
|
676
|
+
__device_forceinline__ unsigned int __uAtomicAdd(unsigned int *p, unsigned int v) { return __nvvm_atom_add_gen_i((int *)p, v); }
|
|
677
|
+
__device_forceinline__ unsigned int __uAtomicAdd_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_add_gen_i((int *)p, v); }
|
|
678
|
+
__device_forceinline__ unsigned int __uAtomicAdd_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_add_gen_i((int *)p, v); }
|
|
679
|
+
__device_forceinline__ unsigned int __uAtomicAnd(unsigned int *p, unsigned int v) { return __nvvm_atom_and_gen_i((int *)p, v); }
|
|
680
|
+
__device_forceinline__ unsigned int __uAtomicAnd_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_and_gen_i((int *)p, v); }
|
|
681
|
+
__device_forceinline__ unsigned int __uAtomicAnd_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_and_gen_i((int *)p, v); }
|
|
682
|
+
__device_forceinline__ unsigned int __uAtomicCAS(unsigned int *p, unsigned int cmp, unsigned int v) { return __nvvm_atom_cas_gen_i((int *)p, cmp, v); }
|
|
683
|
+
__device_forceinline__ unsigned int __uAtomicCAS_block(unsigned int *p, unsigned int cmp, unsigned int v) { return __nvvm_atom_cta_cas_gen_i((int *)p, cmp, v); }
|
|
684
|
+
__device_forceinline__ unsigned int __uAtomicCAS_system(unsigned int *p, unsigned int cmp, unsigned int v) { return __nvvm_atom_sys_cas_gen_i((int *)p, cmp, v); }
|
|
685
|
+
__device_forceinline__ unsigned int __uAtomicDec(unsigned int *p, unsigned int v) { return __nvvm_atom_dec_gen_ui(p, v); }
|
|
686
|
+
__device_forceinline__ unsigned int __uAtomicDec_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_dec_gen_ui(p, v); }
|
|
687
|
+
__device_forceinline__ unsigned int __uAtomicDec_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_dec_gen_ui(p, v); }
|
|
688
|
+
__device_forceinline__ unsigned int __uAtomicExch(unsigned int *p, unsigned int v) { return __nvvm_atom_xchg_gen_i((int *)p, v); }
|
|
689
|
+
__device_forceinline__ unsigned int __uAtomicExch_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_xchg_gen_i((int *)p, v); }
|
|
690
|
+
__device_forceinline__ unsigned int __uAtomicExch_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_xchg_gen_i((int *)p, v); }
|
|
691
|
+
__device_forceinline__ unsigned int __uAtomicInc(unsigned int *p, unsigned int v) { return __nvvm_atom_inc_gen_ui(p, v); }
|
|
692
|
+
__device_forceinline__ unsigned int __uAtomicInc_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_inc_gen_ui(p, v); }
|
|
693
|
+
__device_forceinline__ unsigned int __uAtomicInc_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_inc_gen_ui(p, v); }
|
|
694
|
+
__device_forceinline__ unsigned int __uAtomicMax(unsigned int *p, unsigned int v) { return __nvvm_atom_max_gen_ui(p, v); }
|
|
695
|
+
__device_forceinline__ unsigned int __uAtomicMax_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_max_gen_ui(p, v); }
|
|
696
|
+
__device_forceinline__ unsigned int __uAtomicMax_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_max_gen_ui(p, v); }
|
|
697
|
+
__device_forceinline__ unsigned int __uAtomicMin(unsigned int *p, unsigned int v) { return __nvvm_atom_min_gen_ui(p, v); }
|
|
698
|
+
__device_forceinline__ unsigned int __uAtomicMin_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_min_gen_ui(p, v); }
|
|
699
|
+
__device_forceinline__ unsigned int __uAtomicMin_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_min_gen_ui(p, v); }
|
|
700
|
+
__device_forceinline__ unsigned int __uAtomicOr(unsigned int *p, unsigned int v) { return __nvvm_atom_or_gen_i((int *)p, v); }
|
|
701
|
+
__device_forceinline__ unsigned int __uAtomicOr_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_or_gen_i((int *)p, v); }
|
|
702
|
+
__device_forceinline__ unsigned int __uAtomicOr_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_or_gen_i((int *)p, v); }
|
|
703
|
+
__device_forceinline__ unsigned int __uAtomicXor(unsigned int *p, unsigned int v) { return __nvvm_atom_xor_gen_i((int *)p, v); }
|
|
704
|
+
__device_forceinline__ unsigned int __uAtomicXor_block(unsigned int *p, unsigned int v) { return __nvvm_atom_cta_xor_gen_i((int *)p, v); }
|
|
705
|
+
__device_forceinline__ unsigned int __uAtomicXor_system(unsigned int *p, unsigned int v) { return __nvvm_atom_sys_xor_gen_i((int *)p, v); }
|
|
706
|
+
__device_forceinline__ unsigned int __uhadd(unsigned int a, unsigned int b) { return __nv_uhadd(a, b); }
|
|
707
|
+
__device_forceinline__ double __uint2double_rn(unsigned int a) { return __nv_uint2double_rn(a); }
|
|
708
|
+
__device_forceinline__ float __uint2float_rd(unsigned int a) { return __nv_uint2float_rd(a); }
|
|
709
|
+
__device_forceinline__ float __uint2float_rn(unsigned int a) { return __nv_uint2float_rn(a); }
|
|
710
|
+
__device_forceinline__ float __uint2float_ru(unsigned int a) { return __nv_uint2float_ru(a); }
|
|
711
|
+
__device_forceinline__ float __uint2float_rz(unsigned int a) { return __nv_uint2float_rz(a); }
|
|
712
|
+
__device_forceinline__ float __uint_as_float(unsigned int a) { return __nv_uint_as_float(a); }
|
|
713
|
+
__device_forceinline__ double __ull2double_rd(unsigned long long a) { return __nv_ull2double_rd(a); }
|
|
714
|
+
__device_forceinline__ double __ull2double_rn(unsigned long long a) { return __nv_ull2double_rn(a); }
|
|
715
|
+
__device_forceinline__ double __ull2double_ru(unsigned long long a) { return __nv_ull2double_ru(a); }
|
|
716
|
+
__device_forceinline__ double __ull2double_rz(unsigned long long a) { return __nv_ull2double_rz(a); }
|
|
717
|
+
__device_forceinline__ float __ull2float_rd(unsigned long long a) { return __nv_ull2float_rd(a); }
|
|
718
|
+
__device_forceinline__ float __ull2float_rn(unsigned long long a) { return __nv_ull2float_rn(a); }
|
|
719
|
+
__device_forceinline__ float __ull2float_ru(unsigned long long a) { return __nv_ull2float_ru(a); }
|
|
720
|
+
__device_forceinline__ float __ull2float_rz(unsigned long long a) { return __nv_ull2float_rz(a); }
|
|
721
|
+
__device_forceinline__ unsigned long long __ullAtomicAdd(unsigned long long *p, unsigned long long v) { return __nvvm_atom_add_gen_ll((long long *)p, v); }
|
|
722
|
+
__device_forceinline__ unsigned long long __ullAtomicAdd_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_add_gen_ll((long long *)p, v); }
|
|
723
|
+
__device_forceinline__ unsigned long long __ullAtomicAdd_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_add_gen_ll((long long *)p, v); }
|
|
724
|
+
__device_forceinline__ unsigned long long __ullAtomicAnd(unsigned long long *p, unsigned long long v) { return __nvvm_atom_and_gen_ll((long long *)p, v); }
|
|
725
|
+
__device_forceinline__ unsigned long long __ullAtomicAnd_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_and_gen_ll((long long *)p, v); }
|
|
726
|
+
__device_forceinline__ unsigned long long __ullAtomicAnd_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_and_gen_ll((long long *)p, v); }
|
|
727
|
+
__device_forceinline__ unsigned long long __ullAtomicCAS(unsigned long long *p, unsigned long long cmp, unsigned long long v) { return __nvvm_atom_cas_gen_ll((long long *)p, cmp, v); }
|
|
728
|
+
__device_forceinline__ unsigned long long __ullAtomicCAS_block(unsigned long long *p, unsigned long long cmp, unsigned long long v) { return __nvvm_atom_cta_cas_gen_ll((long long *)p, cmp, v); }
|
|
729
|
+
__device_forceinline__ unsigned long long __ullAtomicCAS_system(unsigned long long *p, unsigned long long cmp, unsigned long long v) { return __nvvm_atom_sys_cas_gen_ll((long long *)p, cmp, v); }
|
|
730
|
+
__device_forceinline__ unsigned long long __ullAtomicExch(unsigned long long *p, unsigned long long v) { return __nvvm_atom_xchg_gen_ll((long long *)p, v); }
|
|
731
|
+
__device_forceinline__ unsigned long long __ullAtomicExch_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_xchg_gen_ll((long long *)p, v); }
|
|
732
|
+
__device_forceinline__ unsigned long long __ullAtomicExch_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_xchg_gen_ll((long long *)p, v); }
|
|
733
|
+
__device_forceinline__ unsigned long long __ullAtomicMax(unsigned long long *p, unsigned long long v) { return __nvvm_atom_max_gen_ull(p, v); }
|
|
734
|
+
__device_forceinline__ unsigned long long __ullAtomicMax_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_max_gen_ull(p, v); }
|
|
735
|
+
__device_forceinline__ unsigned long long __ullAtomicMax_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_max_gen_ull(p, v); }
|
|
736
|
+
__device_forceinline__ unsigned long long __ullAtomicMin(unsigned long long *p, unsigned long long v) { return __nvvm_atom_min_gen_ull(p, v); }
|
|
737
|
+
__device_forceinline__ unsigned long long __ullAtomicMin_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_min_gen_ull(p, v); }
|
|
738
|
+
__device_forceinline__ unsigned long long __ullAtomicMin_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_min_gen_ull(p, v); }
|
|
739
|
+
__device_forceinline__ unsigned long long __ullAtomicOr(unsigned long long *p, unsigned long long v) { return __nvvm_atom_or_gen_ll((long long *)p, v); }
|
|
740
|
+
__device_forceinline__ unsigned long long __ullAtomicOr_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_or_gen_ll((long long *)p, v); }
|
|
741
|
+
__device_forceinline__ unsigned long long __ullAtomicOr_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_or_gen_ll((long long *)p, v); }
|
|
742
|
+
__device_forceinline__ unsigned long long __ullAtomicXor(unsigned long long *p, unsigned long long v) { return __nvvm_atom_xor_gen_ll((long long *)p, v); }
|
|
743
|
+
__device_forceinline__ unsigned long long __ullAtomicXor_block(unsigned long long *p, unsigned long long v) { return __nvvm_atom_cta_xor_gen_ll((long long *)p, v); }
|
|
744
|
+
__device_forceinline__ unsigned long long __ullAtomicXor_system(unsigned long long *p, unsigned long long v) { return __nvvm_atom_sys_xor_gen_ll((long long *)p, v); }
|
|
745
|
+
__device_forceinline__ unsigned int __umul24(unsigned int a, unsigned int b) { return __nv_umul24(a, b); }
|
|
746
|
+
__device_forceinline__ unsigned long long __umul64hi(unsigned long long a, unsigned long long b) { return __nv_umul64hi(a, b); }
|
|
747
|
+
__device_forceinline__ unsigned int __umulhi(unsigned int a, unsigned int b) { return __nv_umulhi(a, b); }
|
|
748
|
+
__device_forceinline__ unsigned int __urhadd(unsigned int a, unsigned int b) { return __nv_urhadd(a, b); }
|
|
749
|
+
__device_forceinline__ unsigned int __usad(unsigned int a, unsigned int b, unsigned int c) { return __nv_usad(a, b, c); }
|
|
750
|
+
|
|
751
|
+
__device_forceinline__ void *memcpy(void *a, const void *b, size_t c) { return __builtin_memcpy(a, b, c); }
|
|
752
|
+
__device_forceinline__ void *memset(void *a, int b, size_t c) { return __builtin_memset(a, b, c); }
|
|
753
|
+
|
|
754
|
+
#if defined(__FAST_MATH__)
|
|
755
|
+
#define __FAST_OR_SLOW(fast, slow) fast
|
|
756
|
+
#else
|
|
757
|
+
#define __FAST_OR_SLOW(fast, slow) slow
|
|
758
|
+
#endif
|
|
759
|
+
|
|
760
|
+
__device_forceinline__ int abs(int a) { return __nv_abs(a); }
|
|
761
|
+
__device_forceinline__ double fabs(double a) { return __nv_fabs(a); }
|
|
762
|
+
__device_forceinline__ double acos(double a) { return __nv_acos(a); }
|
|
763
|
+
__device_forceinline__ float acosf(float a) { return __nv_acosf(a); }
|
|
764
|
+
__device_forceinline__ double acosh(double a) { return __nv_acosh(a); }
|
|
765
|
+
__device_forceinline__ float acoshf(float a) { return __nv_acoshf(a); }
|
|
766
|
+
__device_forceinline__ double asin(double a) { return __nv_asin(a); }
|
|
767
|
+
__device_forceinline__ float asinf(float a) { return __nv_asinf(a); }
|
|
768
|
+
__device_forceinline__ double asinh(double a) { return __nv_asinh(a); }
|
|
769
|
+
__device_forceinline__ float asinhf(float a) { return __nv_asinhf(a); }
|
|
770
|
+
__device_forceinline__ double atan(double a) { return __nv_atan(a); }
|
|
771
|
+
__device_forceinline__ double atan2(double a, double b) { return __nv_atan2(a, b); }
|
|
772
|
+
__device_forceinline__ float atan2f(float a, float b) { return __nv_atan2f(a, b); }
|
|
773
|
+
__device_forceinline__ float atanf(float a) { return __nv_atanf(a); }
|
|
774
|
+
__device_forceinline__ double atanh(double a) { return __nv_atanh(a); }
|
|
775
|
+
__device_forceinline__ float atanhf(float a) { return __nv_atanhf(a); }
|
|
776
|
+
__device_forceinline__ double cbrt(double a) { return __nv_cbrt(a); }
|
|
777
|
+
__device_forceinline__ float cbrtf(float a) { return __nv_cbrtf(a); }
|
|
778
|
+
__device_forceinline__ double ceil(double a) { return __nv_ceil(a); }
|
|
779
|
+
__device_forceinline__ float ceilf(float a) { return __nv_ceilf(a); }
|
|
780
|
+
__device_forceinline__ double copysign(double a, double b) { return __nv_copysign(a, b); }
|
|
781
|
+
__device_forceinline__ float copysignf(float a, float b) { return __nv_copysignf(a, b); }
|
|
782
|
+
__device_forceinline__ double cos(double a) { return __nv_cos(a); }
|
|
783
|
+
__device_forceinline__ float cosf(float a) { return __FAST_OR_SLOW(__nv_fast_cosf, __nv_cosf)(a); }
|
|
784
|
+
__device_forceinline__ double cosh(double a) { return __nv_cosh(a); }
|
|
785
|
+
__device_forceinline__ float coshf(float a) { return __nv_coshf(a); }
|
|
786
|
+
__device_forceinline__ double cospi(double a) { return __nv_cospi(a); }
|
|
787
|
+
__device_forceinline__ float cospif(float a) { return __nv_cospif(a); }
|
|
788
|
+
__device_forceinline__ double cyl_bessel_i0(double a) { return __nv_cyl_bessel_i0(a); }
|
|
789
|
+
__device_forceinline__ float cyl_bessel_i0f(float a) { return __nv_cyl_bessel_i0f(a); }
|
|
790
|
+
__device_forceinline__ double cyl_bessel_i1(double a) { return __nv_cyl_bessel_i1(a); }
|
|
791
|
+
__device_forceinline__ float cyl_bessel_i1f(float a) { return __nv_cyl_bessel_i1f(a); }
|
|
792
|
+
__device_forceinline__ double erf(double a) { return __nv_erf(a); }
|
|
793
|
+
__device_forceinline__ double erfc(double a) { return __nv_erfc(a); }
|
|
794
|
+
__device_forceinline__ float erfcf(float a) { return __nv_erfcf(a); }
|
|
795
|
+
__device_forceinline__ double erfcinv(double a) { return __nv_erfcinv(a); }
|
|
796
|
+
__device_forceinline__ float erfcinvf(float a) { return __nv_erfcinvf(a); }
|
|
797
|
+
__device_forceinline__ double erfcx(double a) { return __nv_erfcx(a); }
|
|
798
|
+
__device_forceinline__ float erfcxf(float a) { return __nv_erfcxf(a); }
|
|
799
|
+
__device_forceinline__ float erff(float a) { return __nv_erff(a); }
|
|
800
|
+
__device_forceinline__ double erfinv(double a) { return __nv_erfinv(a); }
|
|
801
|
+
__device_forceinline__ float erfinvf(float a) { return __nv_erfinvf(a); }
|
|
802
|
+
__device_forceinline__ double exp(double a) { return __nv_exp(a); }
|
|
803
|
+
__device_forceinline__ double exp10(double a) { return __nv_exp10(a); }
|
|
804
|
+
__device_forceinline__ float exp10f(float a) { return __nv_exp10f(a); }
|
|
805
|
+
__device_forceinline__ double exp2(double a) { return __nv_exp2(a); }
|
|
806
|
+
__device_forceinline__ float exp2f(float a) { return __nv_exp2f(a); }
|
|
807
|
+
__device_forceinline__ float expf(float a) { return __nv_expf(a); }
|
|
808
|
+
__device_forceinline__ double expm1(double a) { return __nv_expm1(a); }
|
|
809
|
+
__device_forceinline__ float expm1f(float a) { return __nv_expm1f(a); }
|
|
810
|
+
__device_forceinline__ float fabsf(float a) { return __nv_fabsf(a); }
|
|
811
|
+
__device_forceinline__ double fdim(double a, double b) { return __nv_fdim(a, b); }
|
|
812
|
+
__device_forceinline__ float fdimf(float a, float b) { return __nv_fdimf(a, b); }
|
|
813
|
+
__device_forceinline__ double fdivide(double a, double b) { return a / b; }
|
|
814
|
+
__device_forceinline__ float fdividef(float a, float b) { return __FAST_OR_SLOW(__nv_fast_fdividef(a, b), a / b); }
|
|
815
|
+
__device_forceinline__ double floor(double f) { return __nv_floor(f); }
|
|
816
|
+
__device_forceinline__ float floorf(float f) { return __nv_floorf(f); }
|
|
817
|
+
__device_forceinline__ double fma(double a, double b, double c) { return __nv_fma(a, b, c); }
|
|
818
|
+
__device_forceinline__ float fmaf(float a, float b, float c) { return __nv_fmaf(a, b, c); }
|
|
819
|
+
__device_forceinline__ double fmax(double a, double b) { return __nv_fmax(a, b); }
|
|
820
|
+
__device_forceinline__ float fmaxf(float a, float b) { return __nv_fmaxf(a, b); }
|
|
821
|
+
__device_forceinline__ double fmin(double a, double b) { return __nv_fmin(a, b); }
|
|
822
|
+
__device_forceinline__ float fminf(float a, float b) { return __nv_fminf(a, b); }
|
|
823
|
+
__device_forceinline__ double fmod(double a, double b) { return __nv_fmod(a, b); }
|
|
824
|
+
__device_forceinline__ float fmodf(float a, float b) { return __nv_fmodf(a, b); }
|
|
825
|
+
__device_forceinline__ double frexp(double a, int *b) { return __nv_frexp(a, b); }
|
|
826
|
+
__device_forceinline__ float frexpf(float a, int *b) { return __nv_frexpf(a, b); }
|
|
827
|
+
__device_forceinline__ double hypot(double a, double b) { return __nv_hypot(a, b); }
|
|
828
|
+
__device_forceinline__ float hypotf(float a, float b) { return __nv_hypotf(a, b); }
|
|
829
|
+
__device_forceinline__ int ilogb(double a) { return __nv_ilogb(a); }
|
|
830
|
+
__device_forceinline__ int ilogbf(float a) { return __nv_ilogbf(a); }
|
|
831
|
+
__device_forceinline__ double j0(double a) { return __nv_j0(a); }
|
|
832
|
+
__device_forceinline__ float j0f(float a) { return __nv_j0f(a); }
|
|
833
|
+
__device_forceinline__ double j1(double a) { return __nv_j1(a); }
|
|
834
|
+
__device_forceinline__ float j1f(float a) { return __nv_j1f(a); }
|
|
835
|
+
__device_forceinline__ double jn(int n, double a) { return __nv_jn(n, a); }
|
|
836
|
+
__device_forceinline__ float jnf(int n, float a) { return __nv_jnf(n, a); }
|
|
837
|
+
#if defined(__LP64__)
|
|
838
|
+
__device_forceinline__ long labs(long a) { return __nv_llabs(a); };
|
|
839
|
+
#else
|
|
840
|
+
__device_forceinline__ long labs(long a) { return __nv_abs(a); };
|
|
841
|
+
#endif
|
|
842
|
+
__device_forceinline__ double ldexp(double a, int b) { return __nv_ldexp(a, b); }
|
|
843
|
+
__device_forceinline__ float ldexpf(float a, int b) { return __nv_ldexpf(a, b); }
|
|
844
|
+
__device_forceinline__ double lgamma(double a) { return __nv_lgamma(a); }
|
|
845
|
+
__device_forceinline__ float lgammaf(float a) { return __nv_lgammaf(a); }
|
|
846
|
+
__device_forceinline__ long long llabs(long long a) { return __nv_llabs(a); }
|
|
847
|
+
__device_forceinline__ long long llmax(long long a, long long b) { return __nv_llmax(a, b); }
|
|
848
|
+
__device_forceinline__ long long llmin(long long a, long long b) { return __nv_llmin(a, b); }
|
|
849
|
+
__device_forceinline__ long long llrint(double a) { return __nv_llrint(a); }
|
|
850
|
+
__device_forceinline__ long long llrintf(float a) { return __nv_llrintf(a); }
|
|
851
|
+
__device_forceinline__ long long llround(double a) { return __nv_llround(a); }
|
|
852
|
+
__device_forceinline__ long long llroundf(float a) { return __nv_llroundf(a); }
|
|
853
|
+
__device_forceinline__ double round(double a) { return __nv_round(a); }
|
|
854
|
+
__device_forceinline__ float roundf(float a) { return __nv_roundf(a); }
|
|
855
|
+
__device_forceinline__ double log(double a) { return __nv_log(a); }
|
|
856
|
+
__device_forceinline__ double log10(double a) { return __nv_log10(a); }
|
|
857
|
+
__device_forceinline__ float log10f(float a) { return __nv_log10f(a); }
|
|
858
|
+
__device_forceinline__ double log1p(double a) { return __nv_log1p(a); }
|
|
859
|
+
__device_forceinline__ float log1pf(float a) { return __nv_log1pf(a); }
|
|
860
|
+
__device_forceinline__ double log2(double a) { return __nv_log2(a); }
|
|
861
|
+
__device_forceinline__ float log2f(float a) { return __FAST_OR_SLOW(__nv_fast_log2f, __nv_log2f)(a); }
|
|
862
|
+
__device_forceinline__ double logb(double a) { return __nv_logb(a); }
|
|
863
|
+
__device_forceinline__ float logbf(float a) { return __nv_logbf(a); }
|
|
864
|
+
__device_forceinline__ float logf(float a) { return __FAST_OR_SLOW(__nv_fast_logf, __nv_logf)(a); }
|
|
865
|
+
#if defined(__LP64__)
|
|
866
|
+
__device_forceinline__ long lrint(double a) { return llrint(a); }
|
|
867
|
+
__device_forceinline__ long lrintf(float a) { return __float2ll_rn(a); }
|
|
868
|
+
__device_forceinline__ long lround(double a) { return llround(a); }
|
|
869
|
+
__device_forceinline__ long lroundf(float a) { return llroundf(a); }
|
|
870
|
+
#else
|
|
871
|
+
__device_forceinline__ long lrint(double a) { return (long)rint(a); }
|
|
872
|
+
__device_forceinline__ long lrintf(float a) { return __float2int_rn(a); }
|
|
873
|
+
__device_forceinline__ long lround(double a) { return round(a); }
|
|
874
|
+
__device_forceinline__ long lroundf(float a) { return roundf(a); }
|
|
875
|
+
#endif
|
|
876
|
+
__device_forceinline__ int max(int a, int b) { return __nv_max(a, b); }
|
|
877
|
+
__device_forceinline__ int min(int a, int b) { return __nv_min(a, b); }
|
|
878
|
+
__device_forceinline__ double modf(double a, double *b) { return __nv_modf(a, b); }
|
|
879
|
+
__device_forceinline__ float modff(float a, float *b) { return __nv_modff(a, b); }
|
|
880
|
+
__device_forceinline__ double nearbyint(double a) { return __builtin_nearbyint(a); }
|
|
881
|
+
__device_forceinline__ float nearbyintf(float a) { return __builtin_nearbyintf(a); }
|
|
882
|
+
__device_forceinline__ double nextafter(double a, double b) { return __nv_nextafter(a, b); }
|
|
883
|
+
__device_forceinline__ float nextafterf(float a, float b) { return __nv_nextafterf(a, b); }
|
|
884
|
+
__device_forceinline__ double norm(int dim, const double *t) { return __nv_norm(dim, t); }
|
|
885
|
+
__device_forceinline__ double norm3d(double a, double b, double c) { return __nv_norm3d(a, b, c); }
|
|
886
|
+
__device_forceinline__ float norm3df(float a, float b, float c) { return __nv_norm3df(a, b, c); }
|
|
887
|
+
__device_forceinline__ double norm4d(double a, double b, double c, double d) { return __nv_norm4d(a, b, c, d); }
|
|
888
|
+
__device_forceinline__ float norm4df(float a, float b, float c, float d) { return __nv_norm4df(a, b, c, d); }
|
|
889
|
+
__device_forceinline__ double normcdf(double a) { return __nv_normcdf(a); }
|
|
890
|
+
__device_forceinline__ float normcdff(float a) { return __nv_normcdff(a); }
|
|
891
|
+
__device_forceinline__ double normcdfinv(double a) { return __nv_normcdfinv(a); }
|
|
892
|
+
__device_forceinline__ float normcdfinvf(float a) { return __nv_normcdfinvf(a); }
|
|
893
|
+
__device_forceinline__ float normf(int dim, const float *t) { return __nv_normf(dim, t); }
|
|
894
|
+
__device_forceinline__ double pow(double a, double b) { return __nv_pow(a, b); }
|
|
895
|
+
__device_forceinline__ float powf(float a, float b) { return __nv_powf(a, b); }
|
|
896
|
+
__device_forceinline__ double powi(double a, int b) { return __nv_powi(a, b); }
|
|
897
|
+
__device_forceinline__ float powif(float a, int b) { return __nv_powif(a, b); }
|
|
898
|
+
__device_forceinline__ double rcbrt(double a) { return __nv_rcbrt(a); }
|
|
899
|
+
__device_forceinline__ float rcbrtf(float a) { return __nv_rcbrtf(a); }
|
|
900
|
+
__device_forceinline__ double remainder(double a, double b) { return __nv_remainder(a, b); }
|
|
901
|
+
__device_forceinline__ float remainderf(float a, float b) { return __nv_remainderf(a, b); }
|
|
902
|
+
__device_forceinline__ double remquo(double a, double b, int *c) { return __nv_remquo(a, b, c); }
|
|
903
|
+
__device_forceinline__ float remquof(float a, float b, int *c) { return __nv_remquof(a, b, c); }
|
|
904
|
+
__device_forceinline__ double rhypot(double a, double b) { return __nv_rhypot(a, b); }
|
|
905
|
+
__device_forceinline__ float rhypotf(float a, float b) { return __nv_rhypotf(a, b); }
|
|
906
|
+
__device_forceinline__ double rint(double a) { return __builtin_rint(a); }
|
|
907
|
+
__device_forceinline__ float rintf(float a) { return __builtin_rintf(a); }
|
|
908
|
+
__device_forceinline__ double rnorm(int a, const double *b) { return __nv_rnorm(a, b); }
|
|
909
|
+
__device_forceinline__ double rnorm3d(double a, double b, double c) { return __nv_rnorm3d(a, b, c); }
|
|
910
|
+
__device_forceinline__ float rnorm3df(float a, float b, float c) { return __nv_rnorm3df(a, b, c); }
|
|
911
|
+
__device_forceinline__ double rnorm4d(double a, double b, double c, double d) { return __nv_rnorm4d(a, b, c, d); }
|
|
912
|
+
__device_forceinline__ float rnorm4df(float a, float b, float c, float d) { return __nv_rnorm4df(a, b, c, d); }
|
|
913
|
+
__device_forceinline__ float rnormf(int dim, const float *t) { return __nv_rnormf(dim, t); }
|
|
914
|
+
__device_forceinline__ double rsqrt(double a) { return __nv_rsqrt(a); }
|
|
915
|
+
__device_forceinline__ float rsqrtf(float a) { return __nv_rsqrtf(a); }
|
|
916
|
+
__device_forceinline__ double scalbn(double a, int b) { return __nv_scalbn(a, b); }
|
|
917
|
+
__device_forceinline__ float scalbnf(float a, int b) { return __nv_scalbnf(a, b); }
|
|
918
|
+
__device_forceinline__ double scalbln(double a, long b) {
|
|
919
|
+
if (b > INT_MAX) { return a > 0 ? INFINITY : -HUGE_VAL; }
|
|
920
|
+
if (b < INT_MIN) { return a > 0 ? 0.0 : -0.0; }
|
|
921
|
+
return scalbn(a, (int)b);
|
|
922
|
+
}
|
|
923
|
+
__device_forceinline__ float scalblnf(float a, long b) {
|
|
924
|
+
if (b > INT_MAX) { return a > 0 ? HUGE_VALF : -HUGE_VALF; }
|
|
925
|
+
if (b < INT_MIN) { return a > 0 ? 0.f : -0.f; }
|
|
926
|
+
return scalbnf(a, (int)b);
|
|
927
|
+
}
|
|
928
|
+
__device_forceinline__ double sin(double a) { return __nv_sin(a); }
|
|
929
|
+
__device_forceinline__ void sincos(double a, double *s, double *c) { return __nv_sincos(a, s, c); }
|
|
930
|
+
__device_forceinline__ void sincosf(float a, float *s, float *c) { return __FAST_OR_SLOW(__nv_fast_sincosf, __nv_sincosf)(a, s, c); }
|
|
931
|
+
__device_forceinline__ void sincospi(double a, double *s, double *c) { return __nv_sincospi(a, s, c); }
|
|
932
|
+
__device_forceinline__ void sincospif(float a, float *s, float *c) { return __nv_sincospif(a, s, c); }
|
|
933
|
+
__device_forceinline__ float sinf(float a) { return __FAST_OR_SLOW(__nv_fast_sinf, __nv_sinf)(a); }
|
|
934
|
+
__device_forceinline__ double sinh(double a) { return __nv_sinh(a); }
|
|
935
|
+
__device_forceinline__ float sinhf(float a) { return __nv_sinhf(a); }
|
|
936
|
+
__device_forceinline__ double sinpi(double a) { return __nv_sinpi(a); }
|
|
937
|
+
__device_forceinline__ float sinpif(float a) { return __nv_sinpif(a); }
|
|
938
|
+
__device_forceinline__ double sqrt(double a) { return __nv_sqrt(a); }
|
|
939
|
+
__device_forceinline__ float sqrtf(float a) { return __nv_sqrtf(a); }
|
|
940
|
+
__device_forceinline__ double tan(double a) { return __nv_tan(a); }
|
|
941
|
+
__device_forceinline__ float tanf(float a) { return __nv_tanf(a); }
|
|
942
|
+
__device_forceinline__ double tanh(double a) { return __nv_tanh(a); }
|
|
943
|
+
__device_forceinline__ float tanhf(float a) { return __nv_tanhf(a); }
|
|
944
|
+
__device_forceinline__ double tgamma(double a) { return __nv_tgamma(a); }
|
|
945
|
+
__device_forceinline__ float tgammaf(float a) { return __nv_tgammaf(a); }
|
|
946
|
+
__device_forceinline__ double trunc(double a) { return __nv_trunc(a); }
|
|
947
|
+
__device_forceinline__ float truncf(float a) { return __nv_truncf(a); }
|
|
948
|
+
__device_forceinline__ unsigned long long ullmax(unsigned long long a, unsigned long long b) { return __nv_ullmax(a, b); }
|
|
949
|
+
__device_forceinline__ unsigned long long ullmin(unsigned long long a, unsigned long long b) { return __nv_ullmin(a, b); }
|
|
950
|
+
__device_forceinline__ unsigned int umax(unsigned int a, unsigned int b) { return __nv_umax(a, b); }
|
|
951
|
+
__device_forceinline__ unsigned int umin(unsigned int a, unsigned int b) { return __nv_umin(a, b); }
|
|
952
|
+
__device_forceinline__ double y0(double a) { return __nv_y0(a); }
|
|
953
|
+
__device_forceinline__ float y0f(float a) { return __nv_y0f(a); }
|
|
954
|
+
__device_forceinline__ double y1(double a) { return __nv_y1(a); }
|
|
955
|
+
__device_forceinline__ float y1f(float a) { return __nv_y1f(a); }
|
|
956
|
+
__device_forceinline__ double yn(int a, double b) { return __nv_yn(a, b); }
|
|
957
|
+
__device_forceinline__ float ynf(int a, float b) { return __nv_ynf(a, b); }
|
|
958
|
+
|
|
959
|
+
#undef __FAST_OR_SLOW
|
|
960
|
+
|
|
961
|
+
// Implementation of a subset of <cuda_fp16.h> functionality
|
|
962
|
+
struct __half
|
|
963
|
+
{
|
|
964
|
+
unsigned short u;
|
|
965
|
+
};
|
|
966
|
+
|
|
967
|
+
__device_forceinline__ short __half_as_short(const __half h)
|
|
968
|
+
{
|
|
969
|
+
short i;
|
|
970
|
+
asm("{ mov.b16 %0, %1;}\n" : "=h"(i) : "h"(h));
|
|
971
|
+
return i;
|
|
972
|
+
}
|
|
973
|
+
|
|
974
|
+
__device_forceinline__ __half __short_as_half(const short i)
|
|
975
|
+
{
|
|
976
|
+
__half h;
|
|
977
|
+
asm("{ mov.b16 %0, %1;}\n" : "=h"(h) : "h"(i));
|
|
978
|
+
return h;
|
|
979
|
+
}
|
|
980
|
+
|
|
981
|
+
__device_forceinline__ __half __hadd(const __half a, const __half b)
|
|
982
|
+
{
|
|
983
|
+
__half x;
|
|
984
|
+
asm("{ add.f16 %0, %1, %2;}\n" : "=h"(x) : "h"(a), "h"(b));
|
|
985
|
+
return x;
|
|
986
|
+
}
|
|
987
|
+
|
|
988
|
+
// Implementation of a subset of <cuda_runtime.h> functionality
|
|
989
|
+
__device_forceinline__ bool isfinite(double x) { return __nv_isfinited(x); }
|
|
990
|
+
__device_forceinline__ bool isfinite(float x) { return __nv_finitef(x); }
|
|
991
|
+
|
|
992
|
+
__device_forceinline__ unsigned short atomicCAS(unsigned short *address, unsigned short compare, unsigned short val)
|
|
993
|
+
{
|
|
994
|
+
unsigned short r;
|
|
995
|
+
|
|
996
|
+
asm volatile ("{atom.global.cas.b16 %0,[%1],%2,%3; }\n"
|
|
997
|
+
: "=h"(r)
|
|
998
|
+
: "l"(address), "h"(compare), "h"(val)
|
|
999
|
+
: "memory");
|
|
1000
|
+
|
|
1001
|
+
return r;
|
|
1002
|
+
}
|
|
1003
|
+
|
|
1004
|
+
__device_forceinline__ int atomicCAS(int *address, int compare, int val)
|
|
1005
|
+
{
|
|
1006
|
+
return __iAtomicCAS(address, compare, val);
|
|
1007
|
+
}
|
|
1008
|
+
|
|
1009
|
+
__device_forceinline__ __half atomicAdd(__half *const address, const __half val)
|
|
1010
|
+
{
|
|
1011
|
+
unsigned short* address_as_us = (unsigned short*)address;
|
|
1012
|
+
unsigned short old = *address_as_us;
|
|
1013
|
+
unsigned short assumed;
|
|
1014
|
+
|
|
1015
|
+
do
|
|
1016
|
+
{
|
|
1017
|
+
assumed = old;
|
|
1018
|
+
old = atomicCAS(address_as_us, assumed, __half_as_short(__hadd(val, __short_as_half(assumed))));
|
|
1019
|
+
}
|
|
1020
|
+
while (assumed != old);
|
|
1021
|
+
|
|
1022
|
+
return __short_as_half(old);
|
|
1023
|
+
}
|
|
1024
|
+
|
|
1025
|
+
__device_forceinline__ double atomicAdd(double *const address, const double val)
|
|
1026
|
+
{
|
|
1027
|
+
return __dAtomicAdd(address, val);
|
|
1028
|
+
}
|
|
1029
|
+
|
|
1030
|
+
__device_forceinline__ float atomicAdd(float *const address, const float val)
|
|
1031
|
+
{
|
|
1032
|
+
return __fAtomicAdd(address, val);
|
|
1033
|
+
}
|
|
1034
|
+
|
|
1035
|
+
__device_forceinline__ int atomicAdd(int *const address, const int val)
|
|
1036
|
+
{
|
|
1037
|
+
return __iAtomicAdd(address, val);
|
|
1038
|
+
}
|
|
1039
|
+
|
|
1040
|
+
__device_forceinline__ unsigned int atomicAdd(unsigned int *const address, const unsigned int val)
|
|
1041
|
+
{
|
|
1042
|
+
return __uAtomicAdd(address, val);
|
|
1043
|
+
}
|
|
1044
|
+
|
|
1045
|
+
__device_forceinline__ unsigned int atomicAdd(unsigned long long *const address, const unsigned long long val)
|
|
1046
|
+
{
|
|
1047
|
+
return __ullAtomicAdd(address, val);
|
|
1048
|
+
}
|
|
1049
|
+
|
|
1050
|
+
__device_forceinline__ int atomicMin(int *const address, const int val)
|
|
1051
|
+
{
|
|
1052
|
+
return __iAtomicMin(address, val);
|
|
1053
|
+
}
|
|
1054
|
+
|
|
1055
|
+
__device_forceinline__ int atomicMax(int *const address, const int val)
|
|
1056
|
+
{
|
|
1057
|
+
return __iAtomicMax(address, val);
|
|
1058
|
+
}
|