From 7b91c7befd5df91603afe3d61b6d3823e36d3f7a Mon Sep 17 00:00:00 2001
From: Daniil Fukalov <1671137+dfukalov@users.noreply.github.com>
Date: Wed, 3 Aug 2022 02:12:48 +0300
Subject: [PATCH] Fix "warning: control reaches end of non-void function".
 (#607)

---
 python/src/triton.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/src/triton.cc b/python/src/triton.cc
index 940c5bc0a..f72513395 100644
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -558,16 +558,16 @@ void init_triton_codegen(py::module &&m) {
         }
         if(backend == CUDA)
           return cu_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
-        if(backend == ROCM)
-          return hip_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
+        assert(backend == ROCM);
+        return hip_compile_ttir(name, ir, device, num_warps, num_stages, asm_map, extern_lib_map);
       },
       py::return_value_policy::take_ownership);
   m.def("load_binary", [](backend_t backend, const std::string& name, asm_map_t &asm_map, size_t n_shared_bytes, uint64_t dev){
 	py::gil_scoped_release allow_threads;
         if(backend == CUDA)
           return cu_load_binary(name, asm_map, n_shared_bytes, dev);
-        if(backend == ROCM)
-          return hip_load_binary(name, asm_map, n_shared_bytes, dev);
+        assert(backend == ROCM);
+        return hip_load_binary(name, asm_map, n_shared_bytes, dev);
       }, py::return_value_policy::take_ownership);
 }