Fix "warning: control reaches end of non-void function". (#607)
This commit is contained in:
@@ -558,7 +558,7 @@ void init_triton_codegen(py::module &&m) {
|
|||||||
}
|
}
|
||||||
if(backend == CUDA)
|
if(backend == CUDA)
|
||||||
return cu_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
|
return cu_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
|
||||||
if(backend == ROCM)
|
assert(backend == ROCM);
|
||||||
return hip_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
|
return hip_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
|
||||||
},
|
},
|
||||||
py::return_value_policy::take_ownership);
|
py::return_value_policy::take_ownership);
|
||||||
@@ -566,7 +566,7 @@ void init_triton_codegen(py::module &&m) {
|
|||||||
py::gil_scoped_release allow_threads;
|
py::gil_scoped_release allow_threads;
|
||||||
if(backend == CUDA)
|
if(backend == CUDA)
|
||||||
return cu_load_binary(name, asm_map, n_shared_bytes, dev);
|
return cu_load_binary(name, asm_map, n_shared_bytes, dev);
|
||||||
if(backend == ROCM)
|
assert(backend == ROCM);
|
||||||
return hip_load_binary(name, asm_map, n_shared_bytes, dev);
|
return hip_load_binary(name, asm_map, n_shared_bytes, dev);
|
||||||
}, py::return_value_policy::take_ownership);
|
}, py::return_value_policy::take_ownership);
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user