1# Licensed to the Apache Software Foundation (ASF) under one 2# or more contributor license agreements. See the NOTICE file 3# distributed with this work for additional information 4# regarding copyright ownership. The ASF licenses this file 5# to you under the Apache License, Version 2.0 (the 6# "License"); you may not use this file except in compliance 7# with the License. You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, 12# software distributed under the License is distributed on an 13# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 14# KIND, either express or implied. See the License for the 15# specific language governing permissions and limitations 16# under the License. 17"""Common x86 related utilities""" 18from __future__ import absolute_import as _abs 19import tvm 20 21def get_fp32_len(): 22 mcpu = tvm.target.current_target().mcpu 23 fp32_vec_len = 8 24 if mcpu == 'skylake-avx512' or mcpu == 'cascadelake': 25 fp32_vec_len = 16 26 return fp32_vec_len 27