Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add debug info to fprt runtime calls #1843

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
+563 −111
Open

Add debug info to fprt runtime calls #1843

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
30cc978
Add debug info to fprt runtime calls
ivanradanov Apr 22, 2024
386e432
Also log op name
ivanradanov Apr 22, 2024
21b3fe5
Fix tests
ivanradanov Apr 22, 2024
ff5e1b4
Fix file location
ivanradanov Apr 22, 2024
0fcd564
Fix older llvm vers
ivanradanov Apr 22, 2024
dc65cf2
[Truncate] Handle casts and emit warnings (#1846)
ivanradanov Apr 28, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Loading
Loading status checks…
Also log op name
@ivanradanov
ivanradanov committed Apr 22, 2024
commit 386e432d5b783e15f228346ab8f0134defcd5a6d
30 changes: 20 additions & 10 deletions enzyme/include/enzyme/fprt/mpfr-test.h
View file
Original file line number Diff line number Diff line change
@@ -85,15 +85,17 @@ static __enzyme_fp *__enzyme_fprt_double_to_ptr(double d) {
__ENZYME_MPFR_ATTRIBUTES
double __enzyme_fprt_64_52_get(double _a, int64_t exponent, int64_t significand,
int64_t mode, char *loc) {
printf("%p, %s\n", loc, loc);
if (loc)
printf("%p, %s\n", loc, loc);
__enzyme_fp *a = __enzyme_fprt_double_to_ptr(_a);
return mpfr_get_d(a->v, __ENZYME_MPFR_DEFAULT_ROUNDING_MODE);
}

__ENZYME_MPFR_ATTRIBUTES
double __enzyme_fprt_64_52_new(double _a, int64_t exponent, int64_t significand,
int64_t mode, char *loc) {
printf("%p, %s\n", loc, loc);
if (loc)
printf("%p, %s\n", loc, loc);
__enzyme_fp *a = (__enzyme_fp *)malloc(sizeof(__enzyme_fp));
mpfr_init2(a->v, significand);
mpfr_set_d(a->v, _a, __ENZYME_MPFR_DEFAULT_ROUNDING_MODE);
@@ -103,7 +105,8 @@ double __enzyme_fprt_64_52_new(double _a, int64_t exponent, int64_t significand,
__ENZYME_MPFR_ATTRIBUTES
double __enzyme_fprt_64_52_const(double _a, int64_t exponent,
int64_t significand, int64_t mode, char *loc) {
printf("%p, %s\n", loc, loc);
if (loc)
printf("%p, %s\n", loc, loc);
// TODO This should really be called only once for an appearance in the code,
// currently it is called every time a flop uses a constant.
return __enzyme_fprt_64_52_new(_a, exponent, significand, mode, loc);
@@ -113,7 +116,8 @@ __ENZYME_MPFR_ATTRIBUTES
__enzyme_fp *__enzyme_fprt_64_52_new_intermediate(int64_t exponent,
int64_t significand,
int64_t mode, char *loc) {
printf("%p, %s\n", loc, loc);
if (loc)
printf("%p, %s\n", loc, loc);
__enzyme_fp *a = (__enzyme_fp *)malloc(sizeof(__enzyme_fp));
mpfr_init2(a->v, significand);
return a;
@@ -122,7 +126,8 @@ __enzyme_fp *__enzyme_fprt_64_52_new_intermediate(int64_t exponent,
__ENZYME_MPFR_ATTRIBUTES
void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
int64_t mode, char *loc) {
printf("%p, %s\n", loc, loc);
if (loc)
printf("%p, %s\n", loc, loc);
free(__enzyme_fprt_double_to_ptr(a));
}

@@ -133,7 +138,8 @@ void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
RET __enzyme_fprt_##FROM_TYPE##_##OP_TYPE##_##LLVM_OP_NAME( \
ARG1 a, int64_t exponent, int64_t significand, int64_t mode, \
char *loc) { \
printf("%p, %s\n", loc, loc); \
if (loc) \
printf("%p, %s, %s\n", loc, #LLVM_OP_NAME, loc); \
if (__enzyme_fprt_is_op_mode(mode)) { \
mpfr_t ma, mc; \
mpfr_init2(ma, significand); \
@@ -164,7 +170,8 @@ void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
RET __enzyme_fprt_##FROM_TYPE##_##OP_TYPE##_##LLVM_OP_NAME( \
ARG1 a, ARG2 b, int64_t exponent, int64_t significand, int64_t mode, \
char *loc) { \
printf("%p, %s\n", loc, loc); \
if (loc) \
printf("%p, %s, %s\n", loc, #LLVM_OP_NAME, loc); \
if (__enzyme_fprt_is_op_mode(mode)) { \
mpfr_t ma, mc; \
mpfr_init2(ma, significand); \
@@ -193,7 +200,8 @@ void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
RET __enzyme_fprt_##FROM_TYPE##_##OP_TYPE##_##LLVM_OP_NAME( \
ARG1 a, ARG2 b, int64_t exponent, int64_t significand, int64_t mode, \
char *loc) { \
printf("%p, %s\n", loc, loc); \
if (loc) \
printf("%p, %s, %s\n", loc, #LLVM_OP_NAME, loc); \
if (__enzyme_fprt_is_op_mode(mode)) { \
mpfr_t ma, mb, mc; \
mpfr_init2(ma, significand); \
@@ -225,7 +233,8 @@ void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
TYPE __enzyme_fprt_##FROM_TYPE##_intr_##LLVM_OP_NAME##_##LLVM_TYPE( \
TYPE a, TYPE b, TYPE c, int64_t exponent, int64_t significand, \
int64_t mode, char *loc) { \
printf("%p, %s\n", loc, loc); \
if (loc) \
printf("%p, %s, %s\n", loc, #LLVM_OP_NAME, loc); \
if (__enzyme_fprt_is_op_mode(mode)) { \
mpfr_t ma, mb, mc, mmul, madd; \
mpfr_init2(ma, significand); \
@@ -268,7 +277,8 @@ void __enzyme_fprt_64_52_delete(double a, int64_t exponent, int64_t significand,
bool __enzyme_fprt_##FROM_TYPE##_fcmp_##NAME( \
TYPE a, TYPE b, int64_t exponent, int64_t significand, int64_t mode, \
char *loc) { \
printf("%p, %s\n", loc, loc); \
if (loc) \
printf("%p, %s, %s\n", loc, "fcmp" #NAME, loc); \
if (__enzyme_fprt_is_op_mode(mode)) { \
mpfr_t ma, mb; \
mpfr_init2(ma, significand); \
Loading