-
Notifications
You must be signed in to change notification settings - Fork 49
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.
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
feat[next]: Add support for more datatypes #1786
base: main
Are you sure you want to change the base?
Conversation
…ar_type to avoid duplication
@@ -192,13 +192,13 @@ class UniqueInitializer(DataInitializer): | |||
data containers. | |||
""" | |||
|
|||
start: int = 0 | |||
start: int = 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I might have added a test that relies on start being zero, which is a mistake. Let me know if you see a problem.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please double check the changes I pushed.
src/gt4py/next/program_processors/codegens/gtfn/itir_to_gtfn_ir.py
Outdated
Show resolved
Hide resolved
@@ -318,7 +318,9 @@ def testee(a: int32, b: int32, c: cases.IField) -> cases.IField: | |||
# not inlined | |||
return tmp2 * tmp2 * c | |||
|
|||
cases.verify_with_default_data(cartesian_case, testee, ref=lambda a, b, c: a * b * a * b * c) | |||
cases.verify_with_default_data( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This test erroneously passed before since verify_with_default_data
used 0
for a
which made the entire expression 0.
I checked them and they look good to me. |
This builds on PR#1708 without the
float16
andbfloat16
changes.Add support for
int8, uin8, int16, uint16, uint32
anduint64
.Move builtin definitions from
src/gt4py/next/iterator/ir.py
tosrc/gt4py/next/iterator/builtins.py
.Use ascending integer values in
ScalarKind
-Enum and modify tests respectively.Set
start: int = 1
intests/next_tests/integration_tests/cases.py
to not start initialization from zero as this has the same value as zero-initialized memory and modify tests respectively.