mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
[CI] rename ci and add cron job for amd (#433)
## Summary <!--- This is a required section; please describe the main purpose of this proposed code change. ---> <!--- ## Details This is an optional section; is there anything specific that reviewers should be aware of? ---> ## Testing Done <!--- This is a required section; please describe how this change was tested. ---> <!-- Replace BLANK with your device type. For example, A100-80G-PCIe Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> - Hardware Type: <BLANK> - [ ] run `make test` to ensure correctness - [ ] run `make checkstyle` to ensure code style - [ ] run `make test-convergence` to ensure convergence
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
name: GitHub Actions CI (AMD)
|
||||
name: AMD GPU CI
|
||||
|
||||
on:
|
||||
push:
|
||||
@@ -10,9 +10,13 @@ on:
|
||||
pull_request:
|
||||
branches:
|
||||
- main
|
||||
# paths:
|
||||
# - "src/**"
|
||||
# - "test/**"
|
||||
paths:
|
||||
- "src/**"
|
||||
- "test/**"
|
||||
schedule:
|
||||
# Runs at 00:00 UTC daily
|
||||
- cron: '0 0 * * *'
|
||||
workflow_dispatch: # Enables manual trigger
|
||||
|
||||
concurrency:
|
||||
# This causes it to cancel previous in-progress actions on the same PR / branch,
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
name: Modal GPU CI
|
||||
name: NVIDIA GPU CI
|
||||
|
||||
on:
|
||||
push:
|
||||
@@ -9,10 +9,10 @@ from liger_kernel.ops.cross_entropy import (
|
||||
LigerCrossEntropyFunction,
|
||||
liger_cross_entropy_kernel,
|
||||
)
|
||||
from liger_kernel.ops.utils import is_hip
|
||||
from liger_kernel.transformers.cross_entropy import LigerCrossEntropyLoss
|
||||
from liger_kernel.transformers.functional import liger_cross_entropy
|
||||
from liger_kernel.utils import infer_device
|
||||
from liger_kernel.ops.utils import is_hip
|
||||
|
||||
device = infer_device()
|
||||
set_seed(42)
|
||||
|
||||
@@ -373,7 +373,6 @@ class HFAlignmentLoss:
|
||||
labels: Labels for which to compute the log probabilities. Label tokens with a value of ignore_index are ignored. Shape: (batch_size, sequence_length)
|
||||
average_log_prob: If True, return the average log probability per (non-masked) token. Otherwise, return the sum of the log probabilities of the (non-masked) tokens.
|
||||
is_encoder_decoder: Whether the model is an encoder-decoder model.
|
||||
|
||||
Returns:
|
||||
A tensor of shape (batch_size,) containing the average/sum log probabilities of the given labels under the given logits.
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user