python/tvm/relax/frontend/stablehlo/stablehlo_translator.py (4 lines): - line 273: # TODO (yongwww): Infer the layout automatically - line 386: # TODO (yongwww): Handle mlir.ir.Module with multiple functions - line 399: # TODO (yongwww): handle multiple outputs - line 438: # TODO (yongwww): support the serialized bytecode format of StableHLO python/tvm/relax/frontend/onnx/onnx_frontend.py (3 lines): - line 1759: # TODO (jwfromm) currently only supports constant parameters. - line 3671: # TODO: for now we only really support ai.onnx op set - line 3672: # TODO: handle other namespaces well see https://github.com/apache/tvm/issues/10950 python/tvm/relax/frontend/torch/exported_program_translator.py (3 lines): - line 105: # TODO does not have correctness! - line 106: # TODO we need to store the running mean and variance returned by the - line 185: # TODO figure out why pytorch export passes a list such as src/runtime/hexagon/ops/conv2d_fp16_hvx.cc (3 lines): - line 172: * TODO (quic-sanirudh) Add support for bias add - line 176: * TODO (quic-sanirudh) Add support for relu activation - line 473: // FIXME: Setting bias to zero_block: this works for up to 256 output channels. python/tvm/relax/testing/ast_printer.py (3 lines): - line 222: # TODO: List other attributes? - line 228: # TODO: We may want to print PrimExpr ASTs, but this is a simplification for now - line 267: # TODO: skipping type params and type constraints python/tvm/dlight/gpu/general_reduction.py (2 lines): - line 93: # TODO: fix num_leading_s = 0 case - line 163: # TODO: It's just a workaround to avoid unroll spatial loops, because of the bug of python/tvm/script/ir_builder/tir/triton.py (2 lines): - line 122: # TODO: Support default argument in the kernel - line 123: # TODO: Add specialization for aligned buffer pointers src/runtime/relax_vm/paged_kv_cache.cc (2 lines): - line 1258: // FIXME: if the sender and recver's PP/TP degree do not match, we will need to first - line 1268: // FIXME: if the sender and recver's PP/TP degree do not match, we will need to first python/tvm/ffi/cython/ndarray.pxi (1 line): - line 281: # TODO: Implement python/tvm/topi/gpu/scan.py (1 line): - line 405: # TODO: add support for a prod_scan src/script/printer/doc_printer/base_doc_printer.cc (1 line): - line 66: // FIXME: this only works for ASCII characters. ci/scripts/github/update_branch.py (1 line): - line 168: # TODO: Remove this before landing src/runtime/contrib/nvshmem/init.cc (1 line): - line 61: // FIXME: this is a hack to avoid the issue of NVSHMEM using Multi-process-per-GPU to initialize apps/ios_rpc/tvmrpc/RPCServer.mm (1 line): - line 172: worker_thread_ = nil; // TODO: is it valid? may be better to do that inside NSThread? src/relax/op/distributed/ccl.cc (1 line): - line 33: // FIXME: this is a hack where there's only 1d mesh src/relax/transform/dataflow_inplace.cc (1 line): - line 222: // function constant: give them a fresh index (TODO: we can handle in more detail if this is a src/runtime/opencl/opencl_device_api.cc (1 line): - line 154: /* TODO: the warp size of OpenCL device is not always 1 src/runtime/file_utils.cc (1 line): - line 164: // FIXME: This doesn't check the return code. src/relax/distributed/transform/lower_distir.cc (1 line): - line 132: // FIXME: this is a hack that only works for 1d device mesh src/relax/distributed/transform/lower_global_view_to_local_view.cc (1 line): - line 144: // FIXME: change to use unordered_map (represent dim and sharding spec) web/src/runtime.ts (1 line): - line 202: // TODO: remove soon. Older artifacts do not have this, try-catch for backward compatibility. python/tvm/relax/op/nn/nn.py (1 line): - line 491: # TODO: symbolic shape is not fully supported now rust/tvm-rt/src/object/mod.rs (1 line): - line 45: // TODO we would prefer to blanket impl From/TryFrom ArgValue/RetValue, but we src/relax/op/distributed/distributed.cc (1 line): - line 195: // FIXME: this is a hack where there's only 1d mesh jvm/core/src/main/java/org/apache/tvm/NativeLibraryLoader.java (1 line): - line 40: * TODO: shutdown hook won't work on Windows python/tvm/script/parser/tir/parser.py (1 line): - line 392: # TODO: handle different types of arguments: src/runtime/contrib/coreml/coreml_runtime.mm (1 line): - line 170: // TODO: Support multiple outputs. python/tvm/dlight/gpu/reduction.py (1 line): - line 18: # TODO: combine reduction rule and general reduction rule into one file. ci/scripts/github/github_tvmbot.py (1 line): - line 374: # TODO: Update latest comment in-place if there has been no activity python/tvm/dlight/gpu/gemv.py (1 line): - line 346: # TODO: set LOAD_V_SHARED = False for now python/tvm/ffi/cython/function.pxi (1 line): - line 23: # TODO: Implement python/tvm/script/parser/core/parser.py (1 line): - line 384: # TODO: only the last decorator is parsed