"external/rocm/include/bfloat16_dev.hpp" did not exist on "9aaeacc82b4a6c731cf40d6c450fcc1d695fc94e"
Commit 3a6a31c5 authored by guchaoyang's avatar guchaoyang
Browse files

[Bugfix] Pass pre commit check

parent bbbf4207
......@@ -865,12 +865,12 @@ LayoutMap GemmNode::InferLayout(const LayoutInferArgs &T,
<< "CDNA gemm (FMMA) only supports C in local.fragment scope, got "
<< c_.scope();
if (TargetIsDCU(T.target)) {
auto fragment =
makeGemmFragmentCDCU(m_, n_, m_ / warp_m, n_ / warp_n, c_->dtype.bits());
auto fragment = makeGemmFragmentCDCU(m_, n_, m_ / warp_m, n_ / warp_n,
c_->dtype.bits());
results.Set(c_, fragment->BindThreadRange(thread_range));
} else {
auto fragment =
makeGemmFragmentCCDNA(m_, n_, m_ / warp_m, n_ / warp_n, c_->dtype.bits());
auto fragment = makeGemmFragmentCCDNA(m_, n_, m_ / warp_m, n_ / warp_n,
c_->dtype.bits());
results.Set(c_, fragment->BindThreadRange(thread_range));
}
if (a_.scope() == "shared" || a_.scope() == "shared.dyn") {
......
......@@ -226,11 +226,10 @@ def have_matrixcore(compute_version=None):
return False
@tvm_ffi.register_global_func("tvm_callback_rocm_get_arch", override=True)
def get_rocm_arch(rocm_path="/opt/rocm"):
# @tvm.ffi.register_func("tvm_callback_rocm_get_arch", override=True)
# def get_rocm_arch(rocm_path="/opt/dtk"):
# @tvm.ffi.register_func("tvm_callback_rocm_get_arch", override=True)
# def get_rocm_arch(rocm_path="/opt/dtk"):
"""Utility function to get the AMD GPU architecture
Parameters
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment