[feat] pyfg use avx512 according to cpuinfo (#20)

alibaba · Oct 30, 2024 · 3922def · 3922def
1 parent 0d34ca6
commit 3922def
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -28,7 +28,7 @@ model_config {
 ```
 model_config {
     losses {
-        SoftmaxCrossEntropy {
+        softmax_cross_entropy {
         }
     }
 }

diff --git 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
@@ -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"