Fix for #181 (add dot to CAST list)

https://github.com/NVIDIA/apex/issues/181#issuecomment-507958143

Fix for #181 (add dot to CAST list)
https://github.com/NVIDIA/apex/issues/181#issuecomment-507958143
d6271729 · mcarilli · GitHub · 14fe2b67 · d6271729
Unverified Commit d6271729 authored Jul 03, 2019 by mcarilli Committed by GitHub Jul 03, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

apex/amp/lists/torch_overrides.py apex/amp/lists/torch_overrides.py +1 -0

No files found.
--- a/apex/amp/lists/torch_overrides.py
+++ b/apex/amp/lists/torch_overrides.py
@@ -87,6 +87,7 @@ CASTS = [
    'atan2',
    'cross',
    'bilinear',
+    'dot',

    # Element-wise _or_ tensor-wise math
    'add',