MCPcopy Create free account
hub / github.com/apache/tvm / cuda

Method cuda

python/tvm/rpc/client.py:223–225  ·  view source on GitHub ↗

Construct CUDA GPU device.

(self, dev_id=0)

Source from the content-addressed store, hash-verified

221 return self.device(DLDeviceType.kDLCPU, dev_id)
222
223 def cuda(self, dev_id=0):
224 """Construct CUDA GPU device."""
225 return self.device(DLDeviceType.kDLCUDA, dev_id)
226
227 def cl(self, dev_id=0):
228 """Construct OpenCL device."""

Callers 15

_relax_backendFunction · 0.45
benchmarkFunction · 0.45
get_runtime_deviceFunction · 0.45
has_tensorcoreFunction · 0.45
get_target_compute_versionFunction · 0.45
have_tensorcoreFunction · 0.45
verify_group_gemmFunction · 0.45
test_fp8_e4m3_groupwise_scaled_gemmFunction · 0.45
test_fp8_e4m3_groupwise_scaled_bmmFunction · 0.45
test_tir_triton_integrationFunction · 0.45
test_module_creation_and_instantiation_gpuMethod · 0.45
test_end_to_end_pipeline_gpuMethod · 0.45

Calls 1

deviceMethod · 0.95

Tested by 15

verify_group_gemmFunction · 0.36
test_fp8_e4m3_groupwise_scaled_gemmFunction · 0.36
test_fp8_e4m3_groupwise_scaled_bmmFunction · 0.36
test_tir_triton_integrationFunction · 0.36
test_module_creation_and_instantiation_gpuMethod · 0.36
test_end_to_end_pipeline_gpuMethod · 0.36
test_dlpack_pytorch_to_tvm_conversion_gpuMethod · 0.36
test_dlpack_tvm_to_pytorch_conversion_gpuMethod · 0.36
test_vm_runFunction · 0.36
test_capture_error_is_recoverableFunction · 0.36
test_base_py_module_instantiation_gpuMethod · 0.36
test_call_tir_with_pytorch_tensors_gpuMethod · 0.36