From 3922def30c3650017ce94525af522c9eb9145069 Mon Sep 17 00:00:00 2001
From: Hongsheng Jin <hongsheng.jhs@alibaba-inc.com>
Date: Wed, 30 Oct 2024 18:52:08 +0800
Subject: [PATCH] [feat] pyfg use avx512 according to cpuinfo (#20)

---
 .github/workflows/codestyle_ci.yml | 1 +
 docs/source/models/loss.md         | 2 +-
 requirements/runtime.txt           | 4 ++--
 tzrec/version.py                   | 2 +-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/codestyle_ci.yml b/.github/workflows/codestyle_ci.yml
index a70ca64..1e13d74 100644
--- a/.github/workflows/codestyle_ci.yml
+++ b/.github/workflows/codestyle_ci.yml
@@ -19,4 +19,5 @@ jobs:
         id: run_codestyle_ci
         run: |
           pip install -r requirements.txt
+          git config --global --add safe.directory $(pwd)
           pre-commit run -a
diff --git a/docs/source/models/loss.md b/docs/source/models/loss.md
index 03e219d..0bf1fb0 100644
--- a/docs/source/models/loss.md
+++ b/docs/source/models/loss.md
@@ -28,7 +28,7 @@ model_config {
 ```
 model_config {
     losses {
-        SoftmaxCrossEntropy {
+        softmax_cross_entropy {
         }
     }
 }
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
index ed9ebdd..cbc34db 100644
--- a/requirements/runtime.txt
+++ b/requirements/runtime.txt
@@ -7,8 +7,8 @@ graphlearn @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/graphlearn-1.
 graphlearn @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/graphlearn-1.3.0-cp310-cp310-linux_x86_64.whl ; python_version=="3.10"
 grpcio-tools<1.63.0
 pandas
-pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.2-cp311-cp311-linux_x86_64.whl ; python_version=="3.11"
-pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.2-cp310-cp310-linux_x86_64.whl ; python_version=="3.10"
+pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.4-cp311-cp311-linux_x86_64.whl ; python_version=="3.11"
+pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.4-cp310-cp310-linux_x86_64.whl ; python_version=="3.10"
 scikit-learn
 tensorboard
 torch==2.5.0
diff --git a/tzrec/version.py b/tzrec/version.py
index 86e4ab4..7f5846f 100644
--- a/tzrec/version.py
+++ b/tzrec/version.py
@@ -9,4 +9,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"