Fix "warning: control reaches end of non-void function". (#607)

This commit is contained in:
Daniil Fukalov
2022-08-03 02:12:48 +03:00
committed by GitHub
parent 968f59027e
commit 7b91c7befd

View File

@@ -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);
} }