Skip to content

GitLab

Commit ea5cc27a authored by wenjh's avatar wenjh
Browse files

Update version to das1.8 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent c1a1c04e
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
build_tools/te_version.py
View file @ ea5cc27a
...@@ -7,7 +7,7 @@ import os ...@@ -7,7 +7,7 @@ import os
from pathlib import Path from pathlib import Path
import subprocess import subprocess
DAS_VERSION="1.6" DAS_VERSION="1.8"
def abi_value(): def abi_value():
try: try:
......
qa/L0_pytorch_unittest/test.sh
View file @ ea5cc27a
...@@ -52,6 +52,7 @@ python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_kv_cache.xml $TE ...@@ -52,6 +52,7 @@ python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_kv_cache.xml $TE
python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_hf_integration.xml $TE_PATH/tests/pytorch/test_hf_integration.py || test_fail "test_hf_integration.py" python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_hf_integration.xml $TE_PATH/tests/pytorch/test_hf_integration.py || test_fail "test_hf_integration.py"
mkdir -p $TE_PATH/artifacts/tests/pytorch/test_checkpoint && python $TE_PATH/tests/pytorch/test_checkpoint.py --save-checkpoint all --checkpoint-dir $TE_PATH/artifacts/tests/pytorch/test_checkpoint/ mkdir -p $TE_PATH/artifacts/tests/pytorch/test_checkpoint && python $TE_PATH/tests/pytorch/test_checkpoint.py --save-checkpoint all --checkpoint-dir $TE_PATH/artifacts/tests/pytorch/test_checkpoint/
NVTE_TEST_CHECKPOINT_ARTIFACT_PATH=$TE_PATH/artifacts/tests/pytorch/test_checkpoint python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_checkpoint.xml $TE_PATH/tests/pytorch/test_checkpoint.py || test_fail "test_checkpoint.py" NVTE_TEST_CHECKPOINT_ARTIFACT_PATH=$TE_PATH/artifacts/tests/pytorch/test_checkpoint python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_checkpoint.xml $TE_PATH/tests/pytorch/test_checkpoint.py || test_fail "test_checkpoint.py"
python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_batched_linear.xml $TE_PATH/tests/pytorch/test_batched_linear.py || test_fail "test_batched_linear.py"
python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_fused_router.xml $TE_PATH/tests/pytorch/test_fused_router.py || test_fail "test_fused_router.py" python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_fused_router.xml $TE_PATH/tests/pytorch/test_fused_router.py || test_fail "test_fused_router.py"
if [ "$RET" -ne 0 ]; then if [ "$RET" -ne 0 ]; then
......
tests/cpp/test_common.h
View file @ ea5cc27a
...@@ -17,7 +17,6 @@ ...@@ -17,7 +17,6 @@
#include <cuda_runtime_api.h> #include <cuda_runtime_api.h>
#include <cuda_bf16.h> #include <cuda_bf16.h>
#include <cuda_fp8.h>
#include <cuda_fp16.h> #include <cuda_fp16.h>
#include <cuda_fp8.h> #include <cuda_fp8.h>
#if FP4_TYPE_SUPPORTED #if FP4_TYPE_SUPPORTED
......
tests/pytorch/test_multi_tensor.py
View file @ ea5cc27a
...@@ -257,4 +257,6 @@ def test_multi_tensor_compute_scale_and_scale_inv( ...@@ -257,4 +257,6 @@ def test_multi_tensor_compute_scale_and_scale_inv(
scale_ref, scale_inv_ref, _ = scale_from_amax_tensor( scale_ref, scale_inv_ref, _ = scale_from_amax_tensor(
torch.float32, amax, fp8_dtype, eps=epsilon, pow_2_scales=pow_2_scales torch.float32, amax, fp8_dtype, eps=epsilon, pow_2_scales=pow_2_scales
) )
torch.testing.assert_close(scale, scale_ref, rtol=0, atol=0)
torch.testing.assert_close(scale_inv, scale_inv_ref, rtol=0, atol=0)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment