Skip to content

GitLab

Unverified Commit d6271729 authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Fix for #181 (add dot to CAST list) 

https://github.com/NVIDIA/apex/issues/181#issuecomment-507958143
parent 14fe2b67
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
apex/amp/lists/torch_overrides.py
View file @ d6271729
...@@ -87,6 +87,7 @@ CASTS = [ ...@@ -87,6 +87,7 @@ CASTS = [
'atan2', 'atan2',
'cross', 'cross',
'bilinear', 'bilinear',
'dot',
# Element-wise _or_ tensor-wise math # Element-wise _or_ tensor-wise math
'add', 'add',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment