11 lines
189 B
C
11 lines
189 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <c10/macros/Export.h>
|
||
|
|
|
||
|
|
namespace at::cpu {
|
||
|
|
|
||
|
|
// Detect if CPU support Vector Neural Network Instruction.
|
||
|
|
TORCH_API bool is_cpu_support_vnni();
|
||
|
|
|
||
|
|
} // namespace at::cpu
|