hub / github.com/apache/tvm / test_runtime_builtin_kv_cache_transfer.py

File test_runtime_builtin_kv_cache_transfer.py

tests/python/relax/nvshmem/test_runtime_builtin_kv_cache_transfer.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	# Licensed to the Apache Software Foundation (ASF) under one
2	# or more contributor license agreements. See the NOTICE file
3	# distributed with this work for additional information
4	# regarding copyright ownership. The ASF licenses this file

nothing calls this directly

init_nvshmemFunction · 0.85

test_paged_attention_kv_cache_transferFunction · 0.85

productMethod · 0.80

get_comm_rankFunction · 0.70

set_global_funcFunction · 0.70

create_kv_cacheFunction · 0.70

cudaMethod · 0.45

deviceMethod · 0.45

no test coverage detected