add call_site location (#399)

was missing

melior/src/ir/location.rs

@@ -5,8 +5,9 @@ use crate::{
     utility::print_callback,
 };
 use mlir_sys::{
-    mlirLocationEqual, mlirLocationFileLineColGet, mlirLocationFusedGet, mlirLocationGetContext,
-    mlirLocationNameGet, mlirLocationPrint, mlirLocationUnknownGet, MlirLocation,
+    mlirLocationCallSiteGet, mlirLocationEqual, mlirLocationFileLineColGet, mlirLocationFusedGet,
+    mlirLocationGetContext, mlirLocationNameGet, mlirLocationPrint, mlirLocationUnknownGet,
+    MlirLocation,
 };
 use std::{
     ffi::c_void,
@@ -57,6 +58,11 @@ impl<'c> Location<'c> {
         }
     }
 
+    /// Creates a call site location.
+    pub fn call_site(callee: Location, caller: Location) -> Self {
+        unsafe { Self::from_raw(mlirLocationCallSiteGet(callee.to_raw(), caller.to_raw())) }
+    }
+
     /// Creates an unknown location.
     pub fn unknown(context: &'c Context) -> Self {
         unsafe { Self::from_raw(mlirLocationUnknownGet(context.to_raw())) }
@@ -138,6 +144,13 @@ mod tests {
         Location::name(&context, "foo", Location::unknown(&context));
     }
 
+    #[test]
+    fn call_site() {
+        let context = Context::new();
+
+        Location::call_site(Location::unknown(&context), Location::unknown(&context));
+    }
+
     #[test]
     fn unknown() {
         Location::unknown(&Context::new());