print address

Author: joelberkeley

backend/src/ffi.cpp

@@ -13,7 +13,13 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 */
+#include <iostream>
+
 extern "C" {
+    void print_address(void* ptr) {
+        std::cout << ptr << std::endl;
+    }
+
     int sizeof_int() {
         return sizeof(int);
     }

backend/src/ffi.h

@@ -14,6 +14,8 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 extern "C" {
+    void print_address(void* ptr);  // doesn't belong here?
+
     int sizeof_int();
 
     void set_array_int(int* arr, int idx, int value);

src/Compiler/Xla/Prim/Util.idr

@@ -21,6 +21,10 @@ public export
 libxla : String -> String
 libxla fname = "C:" ++ fname ++ ",libc_xla_extension"
 
+export
+%foreign (libxla "print_address")
+prim__printAddress : GCAnyPtr -> PrimIO ()
+
 export
 %foreign (libxla "sizeof_int")
 sizeofInt : Int

src/Compiler/Xla/Util.idr

@@ -33,6 +33,10 @@ boolToCInt : Bool -> Int
 boolToCInt True = 1
 boolToCInt False = 0
 
+export
+printAddress : GCAnyPtr -> IO ()
+printAddress = primIO . prim__printAddress
+
 public export
 data IntArray : Type where
   MkIntArray : GCPtr Int -> IntArray