MCPcopy
hub / github.com/apache/tvm / rocm

Method rocm

python/tvm/rpc/client.py:239–241  ·  view source on GitHub ↗

Construct ROCm device.

(self, dev_id=0)

Source from the content-addressed store, hash-verified

237 return self.device(DLDeviceType.kDLMetal, dev_id)
238
239 def rocm(self, dev_id=0):
240 """Construct ROCm device."""
241 return self.device(DLDeviceType.kDLROCM, dev_id)
242
243 def ext_dev(self, dev_id=0):
244 """Construct extension device."""

Callers 10

has_matrixcoreFunction · 0.80
have_matrixcoreFunction · 0.80
verifyFunction · 0.80
verify_batch_matmulFunction · 0.80
create_device_targetFunction · 0.80
test_rocm_inf_nanFunction · 0.80
check_rocmFunction · 0.80
test_rocm_warp_shuffleFunction · 0.80
test_rocm_vectorized_expFunction · 0.80
test_export_load_with_fallbackFunction · 0.80

Calls 1

deviceMethod · 0.95

Tested by 8

verifyFunction · 0.64
verify_batch_matmulFunction · 0.64
create_device_targetFunction · 0.64
test_rocm_inf_nanFunction · 0.64
check_rocmFunction · 0.64
test_rocm_warp_shuffleFunction · 0.64
test_rocm_vectorized_expFunction · 0.64
test_export_load_with_fallbackFunction · 0.64