Fix for #302

54b8a852 · mcarilli · GitHub · 4ff153cd · 54b8a852
Unverified Commit 54b8a852 authored May 13, 2019 by mcarilli Committed by GitHub May 13, 2019
Hide 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
@@ -78,6 +78,7 @@ CASTS = [
    'addcmul',
    'atan2',
    'cross',
+    'bilinear',

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