Update IntelTensorFlow_AMX_BF16_Inference.ipynb

Author: YuningQiu

AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_AMX_BF16_Inference/IntelTensorFlow_AMX_BF16_Inference.ipynb

@@ -226,7 +226,7 @@
    "outputs": [],
    "source": [
     "# Reload the model as the bf16 model with AVX512 to compare inference time\n",
-    "os.environ[\"ONEDNN_MAX_CPU_ISA\"] = \"AVX512_BF16\"\n",
+    "os.environ[\"ONEDNN_MAX_CPU_ISA\"] = \"AVX512_CORE_BF16\"\n",
     "tf.config.optimizer.set_experimental_options({'auto_mixed_precision_onednn_bfloat16':True})\n",
     "bf16_model_noAmx = tf.keras.models.load_model('models/my_saved_model_fp32')\n",
     "\n",
@@ -262,7 +262,7 @@
    "outputs": [],
    "source": [
     "# Reload the model as the bf16 model with AMX to compare inference time\n",
-    "os.environ[\"ONEDNN_MAX_CPU_ISA\"] = \"AMX_BF16\"\n",
+    "os.environ[\"ONEDNN_MAX_CPU_ISA\"] = \"AVX512_CORE_AMX\"\n",
     "tf.config.optimizer.set_experimental_options({'auto_mixed_precision_onednn_bfloat16':True})\n",
     "bf16_model_withAmx = tf.keras.models.load_model('models/my_saved_model_fp32')\n",
     "\n",