Skip to content

Commit 80bdfd6

Browse files
malfetfacebook-github-bot
authored andcommitted
Skip Bfloat16 support when building for VSX (pytorch#61630)
Summary: Copy-paste ifdef guard from vec256/vec256.h Probably fixes pytorch#61575 Pull Request resolved: pytorch#61630 Reviewed By: janeyx99 Differential Revision: D29690676 Pulled By: malfet fbshipit-source-id: f6d91eadab74bcbcb1dc9854ae1b98a0dccacd14
1 parent 43a2f7c commit 80bdfd6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
#pragma once
22

33
#include <ATen/cpu/vec/vec256/functional_base.h>
4+
#if !defined(__VSX__) || !defined(CPU_CAPABILITY_VSX)
45
#include <ATen/cpu/vec/vec256/functional_bfloat16.h>
6+
#endif

0 commit comments

Comments
 (0)