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

remove std_ prefix from math library #431

Merged
merged 8 commits into from
Mar 18, 2021
Merged

remove std_ prefix from math library #431

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
59e42ab
remove std_ prefix from math library
rachitnigam Mar 17, 2021
bd58a6e
update test
rachitnigam Mar 17, 2021
640b011
update relay impl
rachitnigam Mar 17, 2021
eb1184f
Merge branch 'master' into rename-math
cgyurgyik Mar 17, 2021
9c6f803
Add prefix and suffix to Relay components.
cgyurgyik Mar 18, 2021
853dfce
Add arity only.
cgyurgyik Mar 18, 2021
92a8d76
Merge branch 'master' into rename-math
cgyurgyik Mar 18, 2021
dc003b6
Remove re.
cgyurgyik Mar 18, 2021
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
Remove re.
  • Loading branch information
@cgyurgyik
cgyurgyik committed Mar 18, 2021
commit dc003b6436277d385952afd489de57098fd23953
8 changes: 3 additions & 5 deletions frontends/relay/dahlia_impl.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,6 @@
from futil.ast import *
from dahlia_utils import *

import re


####################################################################################################
################## Dahlia Implementations for Relay Call Nodes #####################################
Expand Down Expand Up @@ -197,11 +195,11 @@ def max_pool2d(fd: DahliaFuncDef) -> str:
layout = fd.attributes.get_str("layout")
ceil_mode = fd.attributes.get_int("ceil_mode")
assert (
layout == "NCHW"
layout == "NCHW"
), f"""Layout \'{layout}\' is not currently supported for
nn.max_pool2d; please use `NCHW`"""
assert (
ceil_mode == False
ceil_mode == False
), "`ceil_mode` is not currently supported for nn.max_pool2d"

args = res.comp.args
Expand Down Expand Up @@ -440,7 +438,7 @@ def emit_components(func_defs: List[DahliaFuncDef]) -> str:
for func_def in func_defs:
id = func_def.function_id
assert (
id in RelayCallNodes or id in BinaryOps
id in RelayCallNodes or id in BinaryOps
), f"{id} not supported for lowering."

# If the function is a binary operation, use broadcasting.
Expand Down